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

题目链接

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

题目描述

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

解题思路

1. 本题可以用动态规划求解；
2. 设置dp[5][n+1]，其中n为题目中输入的n，dp[i][j]这一二维数组中元素的意义是，以第i个（a,e,i,o,u中）字符结尾的长度为j的字典序字符串的个数；
3. 可以看出，以a结尾的长度为n的字符串只有长度为n-1的“a…a”，以e结尾的长度为n的字符串为以a结尾的和以e结尾的长度为n-1的字符串的总和，…;
4. 3即为状态转移方程，我们组后的结果是dp[0][n]+dp[1][n]+dp[2][n]+dp[3][n]+dp[4][n]。

代码

class Solution {
public:
    int countVowelStrings(int n) {
        vector<vector<int>> dp(5, vector<int>(n+1, 0));
        for(int i = 1; i <= n; ++i)
        {
            dp[0][i] = dp[0][i-1] = 1; 以aa结尾
            dp[1][i] = dp[1][i-1] + dp[0][i-1]; //以e结尾
            dp[2][i] = dp[2][i-1] + dp[1][i-1] + dp[0][i-1]; //以i结尾
            dp[3][i] = dp[3][i-1] + dp[2][i-1] + dp[1][i-1] + dp[0][i-1]; //以o结尾
            dp[4][i] = dp[4][i-1] + dp[3][i-1] + dp[2][i-1] + dp[1][i-1] + dp[0][i-1]; //以u结尾
        }
        int res = 0;
        for(int i = 0; i < 5; i++)
        {
            res += dp[i][n];
        }
        return res;
    }
};