medhi/code.py

code.py

‎‎​

morecode.py

class Solution:
    # you need startIdx for this prob
    def countVowelStrings(self, n: int) -> int:
        vowels = ['a','e','i','o','u']
        result=[]
        def cvs(curr,n,startIdx):
            if n==1:
                result.append(curr)
                
                return 1
            
            nways=0
            for i in range(startIdx,5):
                nways+=cvs(curr+vowels[i],n-1,i)
            
            return nways
        total=0
        
        
        memoize=[[-1 for i in range(5)] for i in range(n+1)]
        def cvsMemo(n,startIdx):
            if memoize[n][startIdx]!=-1:
                return memoize[n][startIdx]
                
            if n==1:
                memoize[n][startIdx]=1
                return 1
            
            nways=0
            for i in range(startIdx,5):
                nways+=cvsMemo(n-1,i)
            
            memoize[n][startIdx]=nways
            return nways
        
        cvs=cvsMemo
        for i,v in enumerate(vowels):
            total+=cvs(n,i)
        # print(memoize)
        return total