1641. 统计字典序元音字符串的数目 几种不同的状态

数学解法

高中学过的隔板方法

dp思路

1. $dp[j][i],j=0,1,2,3,4$末尾的字幕为$j$，长度为$i$的时候我们取得的最小值字符串的数量。
2. 状态转移方程$dp[0][i]=dp[i-1],dp[1][i]=dp[0][i-1]+dp[1][i-1],...,dp[4][i]=dp[0][i-1]+dp[1][i-1]+dp[2][i-1]+dp[3][i-1]$。
3. 可以优化空间复杂度到$O(1)$。
4. 时间复杂度$O(n)$

未优化空间，空间复杂度$O(n)$

class Solution {
public:
    vector<vector<int>>dp;
    int countVowelStrings(int n) {
        if(n<=0){
            return 0;
        }
        dp.resize(5);
        for(int i=0;i<dp.size();i++){
            dp[i].resize(n+1);
        }

        dp[0][1]=dp[1][1]=dp[2][1]=dp[3][1]=dp[4][1]=1;
        int tmp=0;
        for(int i=2;i<=n;i++){
            tmp=0;
            for(int j=0;j<5;j++){
                tmp+=dp[j][i-1];
                dp[j][i]=tmp;
            }
        }
        int res=0;
        for(int i=0;i<5;i++){
            res+=dp[i][n];
        }
       cout<<res<<endl;
       return res;
    }
};

已经优化空间，空间复杂度$O(1)$

class Solution {
public:
    vector<vector<int>>dp;
    int countVowelStrings(int n) {
        if(n<=0){
            return 0;
        }
        dp.resize(5);
        for(int i=0;i<dp.size();i++){
            // dp[i].resize(n+1);
            dp[i].resize(1);
        }

        dp[0][0]=dp[1][0]=dp[2][0]=dp[3][0]=dp[4][0]=1;
        int tmp=0;
        for(int i=2;i<=n;i++){
            tmp=0;
            for(int j=0;j<5;j++){
                tmp+=dp[j][0];
                dp[j][0]=tmp;
            }
        }
        int res=0;
        for(int i=0;i<5;i++){
            res+=dp[i][0];
        }
       cout<<res<<endl;
       return res;
    }
};