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

最新推荐文章于 2023-03-08 16:08:33 发布

花飞雨追

最新推荐文章于 2023-03-08 16:08:33 发布

阅读量410

点赞数

分类专栏： Leetcode 动态规划文章标签： leetcode 动态规划算法

本文链接：https://blog.csdn.net/qq_36258516/article/details/122532004

版权

Leetcode 同时被 2 个专栏收录

142 篇文章 0 订阅

订阅专栏

动态规划

18 篇文章 0 订阅

订阅专栏

题目

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

字符串中的每个字符都应当是小写元音字母（‘a’, ‘e’, ‘i’, ‘o’, ‘u’）
每个元音 ‘a’ 后面都只能跟着 ‘e’
每个元音 ‘e’ 后面只能跟着 ‘a’ 或者是 ‘i’
每个元音 ‘i’ 后面不能再跟着另一个 ‘i’
每个元音 ‘o’ 后面只能跟着 ‘i’ 或者是 ‘u’
每个元音 ‘u’ 后面只能跟着 ‘a’
由于答案可能会很大，所以请你返回模 10^9 + 7 之后的结果。

解题思路

动态规划题，对五种字母分开讨论，这里按顺序下标 0-4 分别对应 a e i o u 结尾的字符串数量。根据题目的条件，可以推出：

a 前面可以是 e i u 。
e 前面可以是 a i 。
i 前面可以是 e o 。
o 前面可以是 i 。
u 前面可以是 i o 。

那么可以给出递推式：

dp[i][0] = dp[i - 1][1] + dp[i - 1][2] + dp[i - 1][4];
dp[i][1] = dp[i - 1][0] + dp[i - 1][2];
dp[i][2] = dp[i - 1][1] + dp[i - 1][3];
dp[i][3] = dp[i - 1][2];
dp[i][4] = dp[i - 1][2] + dp[i - 1][3];

那么可以实现代码。

代码

class Solution {
    public static final int MOD = 1000000000 + 7;
    public static final int MAX_LENGTH = 20000 + 50;
    public static long[][] dp = new long[MAX_LENGTH][5];

    public static void initDp(int n) {
        for (int i = 0; i < 5; i++) dp[0][i] = 1L;
        for (int i = 1; i < n; i++) {
            dp[i][0] = ((dp[i - 1][1] + dp[i - 1][2]) % MOD + dp[i - 1][4]) % MOD;
            dp[i][1] = (dp[i - 1][0] + dp[i - 1][2]) % MOD;
            dp[i][2] = (dp[i - 1][1] + dp[i - 1][3]) % MOD;
            dp[i][3] = dp[i - 1][2];
            dp[i][4] = (dp[i - 1][2] + dp[i - 1][3]) % MOD;
        }
    }

    public int countVowelPermutation(int n) {
        initDp(n);
        long ans = 0;
        for (int i = 0; i < 5; i++) {
            ans = (ans + dp[n - 1][i]) % MOD;
        }
        return (int) ans;
    }
}