[每日一题]134：统计元音字母序列的数目

---

题目描述

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

题解思路

Ps： 这可能是最简单的 Hard 题了

1. 定义a,e,i,o,u分别为以该字母结尾的序列的个数，然后遍历n-1次，每次都更新这五个变量的值
2. 更新的规则在于，目前长度为n的以a结尾的序列，是由长度为n-1的由e,u和i结尾的序列增加一个a得到的，所以a = e + u + i
3. 以此类推可得另外五个
4. 在每轮更新的时候都取模，比更新完再取模速度快很多

代码如下：

class Solution {
public:
    int countVowelPermutation(int n) {
        const int MOD = 1e9 + 7;
        long long a = 1, e = 1, i = 1, o = 1, u = 1;
        for (int k = 2; k <= n; ++k) {
            long long aa = e % MOD;
            long long ee = (a + i) % MOD;
            long long ii = (a + e + o + u) % MOD;
            long long oo = (i + u) % MOD;
            long long uu = a % MOD;
            a = aa;
            e = ee;
            i = ii;
            o = oo;
            u = uu;
        }
        return (a + e + i + o + u) % MOD;
    }
};