用类似二进制的方法进行循坏

解题思路:
用序列放置各个位上的数字再进行判断
注意事项:
有没有大佬帮忙优化一下，时间有点超限
参考代码:

N=int(input())

K=int(input())

a=[1]*(N)

for i in range(1,N):

    a[i]=0

x=N-1

num=0

f='true'

sum=(K-1)*pow(K,N-1)

for j in range(sum):

    for i in range(N-1):

        if a[i]==0 and a[i+1]==0:

            f='flase'

    if f=='true':

        num+=1

    a[x]+=1

    while a[x]==K:

        a[x-1]+=1

        a[x]=0

        x-=1

    f='true'

    x=N-1

print(num)