Advertisement
Guest User

Untitled

a guest
Nov 18th, 2017
92
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.23 KB | None | 0 0
  1. from functools import lru_cache
  2. n = int(input())
  3. k = int(input())
  4. @lru_cache(1000)
  5. def f(n):
  6. if n == 1:
  7. return k - 1
  8. if n == 2:
  9. return (k - 1) * k
  10.  
  11. return (f(n - 2) + f(n - 1)) * (k - 1)
  12. print(f(n))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement