My code:
n, k = map(int, input().split())
zero = [0, k - 1]
one = [0, 0]
for i in range(2, n + 1):
    zero.append((zero[i - 1] + one[i - 1]) * (k - 1))
    one.append(zero[i - 1])
print(zero[n] + one[n])

It passes all tests I can come up with on local machine but gets RE1 and I can't get what's the problem, help pls.

you should change first line of code to n = input(), k = input()