【leetcode】统计字典序元音字符串的数目（dp，3种写法）

给你一个整数 n，请返回长度为 n 、仅由元音 (a, e, i, o, u) 组成且按 字典序排列 的字符串数量。

字符串 s 按 字典序排列 需要满足：对于所有有效的 i，s[i] 在字母表中的位置总是与 s[i+1] 相同或在 s[i+1] 之前。

示例 1：

输入：n = 1
输出：5
解释：仅由元音组成的 5 个字典序字符串为 [“a”,“e”,“i”,“o”,“u”]
示例 2：

输入：n = 2
输出：15
解释：仅由元音组成的 15 个字典序字符串为
[“aa”,“ae”,“ai”,“ao”,“au”,“ee”,“ei”,“eo”,“eu”,“ii”,“io”,“iu”,“oo”,“ou”,“uu”]
注意，“ea” 不是符合题意的字符串，因为 ‘e’ 在字母表中的位置比 ‘a’ 靠后
示例 3：

输入：n = 33
输出：66045

提示：

1 <= n <= 50

链接：https://leetcode-cn.com/problems/count-sorted-vowel-strings

思路分析：

第一种状态：
dp[i][j]表示长度为 i, 以 j 结尾的方案总数，则返回值为dp[n][0…4]
假设长度为2，以 i 结尾，可以有 ai, ei, ii
那么可以分析出 dp[i][j] = dp[i-1][0…j].
初始状态：dp[1][0…4] = 1

class Solution {
public:
    int countVowelStrings(int n) {
        int dp[n+1][5], ans = 0;
        memset(dp,0,sizeof(dp));
        for(int i = 0;i < 5;i++) dp[1][i] = 1;
        if(n == 1) return 5;
        for(int i = 2;i <= n;i++)
        {
            for(int j = 0;j < 5;j++)
            {
                for(int k = 0;k <= j;k++)
                {
                    dp[i][j] += dp[i-1][k];
                }
                if(i == n) ans += dp[i][j];
            }
        }
        return ans;
    }
};

对于这样的代码，时间复杂度是O(n^3)，不是最简便的，因为有些地方可以优化。
我们观察到dp[i][j] = dp[i-1][0…j]，[0…j]是通过多用一个循环实现的。但这个其实就是前缀和。我们可以定义一个sum[i][j]表示dp[i][0…j]的总和，最后返回sum[n][4].
那么dp[i][j] = sum[i-1][j]，sum[i][j] = sum[i][j-1]+dp[i][j]. 更新当前dp的同时更新dp的前缀和。

class Solution {
public:
    int countVowelStrings(int n) {
        int dp[n+1][5], ans = 0, sum[n+1][5];
        memset(dp,0,sizeof(dp));
        memset(sum,0,sizeof(sum));
        for(int i = 0;i < 5;i++) dp[1][i] = 1;
        sum[1][0] = 1;
        for(int i = 1;i < 5;i++) sum[1][i] = sum[1][i-1]+dp[1][i-1];
        if(n == 1) return 5;
        for(int i = 2;i <= n;i++)
        {
            for(int j = 0;j < 5;j++)
            {
                dp[i][j] = sum[i-1][j];
                if(j == 0) sum[i][j] = dp[i][j];
                else sum[i][j] = sum[i][j-1]+dp[i][j];
            }
        }
        return sum[n][4];
    }
};

再优化一下。其实对于sum这种辅助数组来说也可以是不需要的。
我们将状态定义为dp[i][j]表示长度为 i ，[0…j] 的方案总数，最终求dp[n][4].
对样例["aa","ae","ai","ao","au","ee","ei","eo","eu","ii","io","iu","oo","ou","uu"]分析：
假设i = 2, j = 2，结果有：ai,ei,ii,aa,ae,ee.
我们发现，ai,ei,ii就是dp[i-1][2]，aa就是dp[i-1][0]，ae,ee就是dp[i][1].
那么状态转移方程可以写成：
dp[i][j] = dp[i-1][j]+dp[i][j-1].

//dp[i][j]表示长度为i，[0...j]的方案总数，最终求dp[n][4]
//dp[i][j] = dp[i-1][j] + dp[i][j-1]
//初始化：dp[0][0...4] = 1;

class Solution {
public:
    int countVowelStrings(int n) {
        int dp[n+1][5];
        memset(dp,0,sizeof(dp));
        for(int i = 0;i < 5;i++) dp[0][i] = 1;
        for(int i = 1;i <= n;i++)
        {
            for(int j = 0;j < 5;j++)
            {
                dp[i][j] = dp[i-1][j];
                if(j > 0) dp[i][j] += dp[i][j-1];
            }
        }
        return dp[n][4];
    }
};