Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from functools import lru_cache
- n = int(input())
- k = int(input())
- @lru_cache(1000)
- def f(n):
- if n == 1:
- return k - 1
- if n == 2:
- return (k - 1) * k
- return (f(n - 2) + f(n - 1)) * (k - 1)
- print(f(n))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement