LeetCode·每日一题·1641. 统计字典序元音字符串的数目·动态规划

作者：小迅
链接：https://leetcode.cn/problems/count-sorted-vowel-strings/solutions/2196589/dong-tai-gui-hua-zhu-shi-chao-ji-xiang-x-2c4b/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。

题目

 

思路

题意 -> 给定一个长度为 n 的字符串，请按字典序排列向 字符串中填充 元音字符，返回不同字符串的个数。

动态规划

定义dp[n][5]数组

• 其中分别使用数字 0，1，2，3，4 代表元音字符 ‘a’，‘e’，‘i’，‘o’，‘u’。记 dp[i][j] 表示长度为 i+1，以 j 结尾的按字典序排列的字符串数量，
• 那么动态转移方程为 dp[n][j] = dp[n-1][j-1]，其中 j 为 0-j的累和，代表的含义为当前字符 j 的前面可以为 0 -- j-1的任意一个字符。
• 枚举方向为从头到后。
• 初始化应该为 1，因为当前字符的选择只与上一个字符有关。

代码注释超级详细

代码

int countVowelStrings(int n) {
    int dp[n][5];
    for (int i = 0; i < 5; ++i) dp[0][i] = 1;//初始化
    for (int i = 1; i < n; ++i) {
        dp[i][0] = dp[i - 1][0];    // a前只能是a
        dp[i][1] = dp[i - 1][0] + dp[i - 1][1];     // b前可以是a和b
        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];
    }
    int ret = 0;
    for (int i = 0; i < 5; i++) {//统计最后字符串个数
        ret += dp[n-1][i];
    }
    return ret;
}


作者：小迅
链接：https://leetcode.cn/problems/count-sorted-vowel-strings/solutions/2196589/dong-tai-gui-hua-zhu-shi-chao-ji-xiang-x-2c4b/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。