From 3333839302c05245b35c23241d8d22ef495ba800 Mon Sep 17 00:00:00 2001 From: terminaldweller Date: Sun, 7 Aug 2022 08:57:12 +0430 Subject: update --- 1220/main.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100755 1220/main.py (limited to '1220/main.py') diff --git a/1220/main.py b/1220/main.py new file mode 100755 index 0000000..8d6b919 --- /dev/null +++ b/1220/main.py @@ -0,0 +1,31 @@ +#!/usr/bin/env python3 +# not my solution + + +class Solution: + def countVowelPermutation(self, n: int) -> int: + kMod = int(1e9) + 7 + dp = {"a": 1, "e": 1, "i": 1, "o": 1, "u": 1} + + for _ in range(n - 1): + newDp = { + "a": dp["e"] + dp["i"] + dp["u"], + "e": dp["a"] + dp["i"], + "i": dp["e"] + dp["o"], + "o": dp["i"], + "u": dp["i"] + dp["o"], + } + dp = newDp + + return sum(dp.values()) % kMod + + +def main(): + solution = Solution() + print(solution.countVowelPermutation(1)) + print(solution.countVowelPermutation(2)) + print(solution.countVowelPermutation(5)) + + +if __name__ == "__main__": + main() -- cgit v1.2.3