Count Sorted Vowel Strings LeetCode solution in Python

1
class Solution:
2
    def countVowelStrings(self, n: int) -> int:
3
        dp = [[0] * 6 for _ in range(n + 1)]
4
        for i in range(1, 6):
5
            dp[1][i] = i
6

7
        for i in range(2, n + 1):
8
            dp[i][1] = 1
9
            for j in range(2, 6):
10
                dp[i][j] = dp[i][j - 1] + dp[i - 1][j]
11

12
        return dp[n][5]