Python程序设计题解【蓝桥杯官网题库】 DAY13-算法训练

试题 算法训练 K好数

资源限制
时间限制：1.0s 内存限制：256.0MB
问题描述
如果一个自然数N的K进制表示中任意的相邻的两位都不是相邻的数字，那么我们就说这个数是K好数。求L位K进制数中K好数的数目。例如K = 4，L = 2的时候，所有K好数为11、13、20、22、30、31、33 共7个。由于这个数目很大，请你输出它对1000000007取模后的值。

输入格式
输入包含两个正整数，K和L。

输出格式
输出一个整数，表示答案对1000000007取模后的值。
样例输入
4 2
样例输出
7
数据规模与约定
对于30%的数据，KL <= 106；

对于50%的数据，K <= 16， L <= 10；

对于100%的数据，1 <= K,L <= 100。

while True:
    try:
         n = list(map(int, input().split()))
         k, l = n[0], n[1]
         mod = 1000000007
         result = 0
         dp = [[0 for i in range(l+1)] for j in range(k)]
         #if l == 0 or l == 1:
             #print(0)
         #else:
         for w in range(k):
             dp[w][1] = 1
         for y in range(2, l + 1):
             for x in range(k):
                 for z in range(k):
                     if z != x+1 and z != x-1:
                         dp[x][y] = (dp[x][y]%mod + dp[z][y-1]%mod)%mod
         for i in range(1,k):
             result = result%mod + dp[i][l]%mod
         print(result % mod)
    except:
        break

来源：CSDN

作者：理想007

链接：https://blog.csdn.net/weixin_43838785/article/details/104356953