1641. 统计字典序元音字符串的数目

1641. 统计字典序元音字符串的数目

难度中等

给你一个整数 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 

思路：这道题刚拿到可能会想到用组合数学，比如第一位是a，那么第二位可以用a、e、i、o、u，也就是5个，然而在考虑第三位的时候，取决于第二位具体的字符，所以其实没有办法用组合数学。现在我们换个思路，要记录状态，我们用dp[i][j]表示第i位选择了第j个元音字符的字符串数量，那么根据题意，如果我们第i位选择了第j个元音字符，则第i-1位必须满足只能选择第0-j个元音字符中的一个，即有如下状态转移方程

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