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