Python｜Leetcode《1220》｜统计元音字母序列的数目

Posted 2022-12-10 二哥不像程序员

专栏《LeetCode｜一刷到底》
打卡每天leetcode精选每日一题（尽量不断更！）

点击关注不迷路!!!

一、题目描述

• 题目：统计元音字母序列的数目
• 难度：困难
• 地址：《统计元音字母序列的数目》
• 描述：给你一个整数 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$

二、题目解析

此题虽为难题，但是本质上可以看作是动态规划问题的一般形态。
初始形态为字符串长度为1的时候的形态（n=1），后续随着n的增大，我们不断的替换每一层的字符中有几个a、e、i、o、u即可，例如第二层的字符中，在第一层字符e、i、u的后面会出现字符a。图解如下：

不同颜色的连线代表第二层中的各个字符会由上一层中的哪些字符得到，上一层中有几个这样的字符数量就为几，后面依次类推即可。

三、解题代码

解法

class Solution:
    def countVowelPermutation(self, n: int) -> int:
        a, e, i, o, u = 1, 1, 1, 1, 1
        for _ in range(n - 1):
            a, e, i, o, u = e + u + i, a + i, o + e, i, i + o
        return (a + e + i + o + u) % (10**9 + 7)