【Leetcode】1220. Count Vowels Permutation

题目地址：

https://leetcode.com/problems/count-vowels-permutation/description/

要求构造一个长$n$的只含五个小写元音字符的字符串，规则如下：
1、'a后只能跟'e'；
2、'e'后只能跟'a'或者'i'；
3、'i'后不能跟'i'；
4、'o'后只能跟'i'或者'u'；
5、'u'后只能跟'a'。
问能构造出多少种不同的字符串。

设$f[e][n]$为以字母$e$结尾的长度为$n$的不同字符串有多少个。线性递推一下即可。最后返回$\sum f[.][n]$。代码如下：

class Solution {
 public:
  int countVowelPermutation(int n) {
    const int MOD = 1e9 + 7;
    vector<vector<int>> f(5, vector<int>(n + 1));
    for (int i = 0; i < 5; i++) f[i][1] = 1;
    vector<int> g[5] = {{1}, {0, 2}, {0, 1, 3, 4}, {2, 4}, {0}};
    for (int i = 2; i <= n; i++)
      for (int j = 0; j < 5; j++)
        for (int ne : g[j]) f[ne][i] = (f[ne][i] + f[j][i - 1]) % MOD;

    int res = 0;
    for (int i = 0; i < 5; i++) res = (res + f[i][n]) % MOD;
    return res;
  }
};

时空复杂度$O(n)$。