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

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

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

/*动态规划方法*/
class Solution {
public:
    int countVowelPermutation(int n)
    {
        long long mod = 1e9 + 7;
        vector<long long> dp(5, 1);
        vector<long long> ndp(5);

        for (int i = 1; i < n; i++)
        {
            ndp[0] = (dp[1] + dp[2] + dp[4]) % mod;
            ndp[1] = (dp[0] + dp[2]) % mod;
            ndp[2] = (dp[1] + dp[3]) % mod;
            ndp[3] = (dp[2]) % mod;
            ndp[4] = (dp[2] + dp[3]) % mod;
            dp = ndp;
        }

        return accumulate(dp.begin(), dp.end(), 0LL) % mod;
    }
};

/*快速幂矩阵*/
class Solution {
public:
    vector<vector<long long>> matrixMultiply(const vector<vector<long long>>& M, const vector<vector<long long>>& N, long long mod)
    {
        vector<vector<long long>> resM(M.size(), vector<long long>(N[0].size(), 0));
        for (int i = 0; i < M.size(); i++)
        {
            for (int j = 0; j < N[0].size(); j++)
            {
                for (int k = 0; k < M[0].size(); k++)
                {
                    resM[i][j] = (resM[i][j] + M[i][k] * N[k][j]) % mod;
                }
            }
        }
        return resM;
    }

    vector<vector<long long>> fastPowMatrix(const vector<vector<long long>>& M, int n, long long mod)
    {
        vector<vector<long long>> resM(M.size(), vector<long long>(M.size(), 0));
        vector<vector<long long>> curM = M;

        for (int i = 0; i < M.size(); i++)
        {
            resM[i][i] = 1;
        }

        for (int i = n; i != 0; i >>= 1)
        {
            if (i & 1)
            {
                resM = matrixMultiply(curM, resM, mod);
            }
            curM = matrixMultiply(curM, curM, mod);
        }

        return resM;
    }

    int countVowelPermutation(int n)
    {
        long long mod = 1e9 + 7;
        vector<vector<long long>> M = {
            {0, 1, 0, 0, 0},
            {1, 0, 1, 0, 0},
            {1, 1, 0, 1, 1},
            {0, 0, 1, 0, 1},
            {1, 0, 0, 0, 0}};
        
        vector<vector<long long>> resM = fastPowMatrix(M, n - 1, mod);
        
        long long res = 0;
        for (auto &&row : resM)
        {
            res = (res + accumulate(row.begin(), row.end(), 0LL)) % mod;
        }
        return res;
    }
};