题目描述:给你一个整数 n,请返回长度为 n 、仅由元音 (a, e, i, o, u) 组成且按 字典序排列 的字符串数量。
字符串 s 按 字典序排列 需要满足:对于所有有效的 i,s[i] 在字母表中的位置总是与 s[i+1] 相同或在 s[i+1] 之前。
解题思路:用dp[i][j]
表示以i结尾的长度为j的字符串,他的解全部可以由长度为j-1的字符串递推过来,为长度为j-1且结尾字符的字母顺序在i前面或者和i相等,然后同样的方法写出dp[i+1][j]
的递推式,两式相减可以得到简化的递推式,dp[i+1][j]=dp[i][j]+dp[i+1][j-1]
,这个递推式也可以理解为,以字母i+1
结尾的长度为j的字符串可以由两部分得到,(1)将长度为j且结尾字符为i的字符串的结尾字符都替换为i+1,即dp[i][j]
,(2)结尾字符为i+1,长度为j-1末尾加上一个i+1得到,即dp[i+1[j-1
,代码如下:
class Solution:
def countVowelStrings(self, n: int) -> int:
dp = [[1] * n for _ in range(5)]
for i in range(1, 5):
for j in range(1, n):
dp[i][j] = dp[i-1][j] + dp[i][j-1]
return sum([d[-1] for d in dp])