LeetCode第157场周赛：5216. 统计元音字母序列的数目（动态规划）

5216. 统计元音字母序列的数目

 显示英文描述 

我的提交返回竞赛

• 用户通过次数222
• 用户尝试次数275
• 通过次数227
• 提交次数432
• 题目难度Hard

给你一个整数 n，请你帮忙统计一下我们可以按下述规则形成多少个长度为 n 的字符串：

• 字符串中的每个字符都应当是小写元音字母（'a', 'e', 'i', 'o', 'u'）
• 每个元音 'a' 后面都只能跟着 'e'
• 每个元音 'e' 后面只能跟着 'a' 或者是 'i'
• 每个元音 'i' 后面 不能 再跟着另一个 'i'
• 每个元音 'o' 后面只能跟着 'i' 或者是 'u'
• 每个元音 'u' 后面只能跟着 'a'

由于答案可能会很大，所以请你返回 模 10^9 + 7 之后的结果。

 

示例 1：

输入：n = 1
输出：5
解释：所有可能的字符串分别是："a", "e", "i" , "o" 和 "u"。

示例 2：

输入：n = 2
输出：10
解释：所有可能的字符串分别是："ae", "ea", "ei", "ia", "ie", "io", "iu", "oi", "ou" 和 "ua"。

示例 3：

输入：n = 5
输出：68

 

提示：

• 1 <= n <= 2 * 10^4

思路：设dp[i][j]表示以第j个元音字符结尾的方案数。。。。

想不懂为啥是个hard题

class Solution {
	int mod = 1000000007;
    public int countVowelPermutation(int n) {
        int[][] dp=new int[n+1][5];
        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++)
        		for(int h=0;h<5;h++)
        		{
        			if(h==0 && j!=1 || h==1 && j!=0 && j!=2 || h==2 && j==2) 
        				continue;
        			if(h==3 && j!=2 && j!=4 || h==4 && j!=0)
        				continue;
        			dp[i][j]=(dp[i][j]+dp[i-1][h])%mod;
        		}
        int ans=0;
        for(int i=0;i<5;i++)
        	ans=(ans+dp[n-1][i])%mod;
        return ans;
    }
}