LeetCode1220 统计元音字母序列的数目[动态规划] HERODING的LeetCode之路

最新推荐文章于 2024-09-13 18:11:35 发布

HERODING77

最新推荐文章于 2024-09-13 18:11:35 发布

阅读量182

点赞数 1

分类专栏： LeetCode 文章标签： leetcode 动态规划算法 c++ 数据结构

本文链接：https://blog.csdn.net/HERODING23/article/details/122506163

版权

LeetCode 专栏收录该内容

733 篇文章 6 订阅

订阅专栏

在这里插入图片描述解题思路：
一道非常简单的动态规划题目，因为需要考虑的情况都是常数级别，所以只需要定义5个变量存储到第k个位置时，以任意变量代表的原因形成的字符串个数，注意每次更新要取模，最后返回所有元音数量的和，代码如下：

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