给你一个整数 n,请你帮忙统计一下我们可以按下述规则形成多少个长度为 n 的字符串:
字符串中的每个字符都应当是小写元音字母('a', 'e', 'i', 'o', 'u')
每个元音 'a' 后面都只能跟着 'e'
每个元音 'e' 后面只能跟着 'a' 或者是 'i'
每个元音 'i' 后面 不能 再跟着另一个 'i'
每个元音 'o' 后面只能跟着 'i' 或者是 'u'
每个元音 'u' 后面只能跟着 'a'
由于答案可能会很大,所以请你返回 模 10^9 + 7 之后的结果。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/count-vowels-permutation
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
以a,e,i,o,u为开头的字符串数量,然后每次滚动更新当前字符串数量
class Solution {
public int countVowelPermutation(int n) {
long a=1,e=1,i=1,o=1,u=1;//定义字符串开头的默认值
int k;
int m=1000000007;
for(k=1;k<n;k++)//循环n-1次,第一次循环为默认值,所以循环n-1次
{
long aa=e%m;//a后面只能跟e,以e的字符串数量模除不可超越的值,下面依次类推
long ee=(a+i)%m;
long ii=(a+e+o+u)%m;
long oo=(i+u)%m;
long uu=a%m;
a=aa;
e=ee;
i=ii;
o=oo;
u=uu;
}
return (int)((a+e+i+o+u)%m);//返回各字符开头的字符串数量
}
}