算法刷题系列（四）蓝桥杯python算法训练2

- 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。

• 解答程序

k, l = list(map(int, input().split()))
s = [0 for i in range(l + 1)]
a = [[0 for i in range(l)] for i in range(k)]
# a[x][y]表示最高位为x，剩余长度为y的K好数数目
if l == 1:
    if k == 1:
        print(0)
    else:
        print(k)
else:
    for i in range(k):
        a[i][0] = 1
    s[1] = k 
    for  j in range(1,l):
        for i in range(k):
            if i == 0:
                a[0][j] = (s[j] - a[1][j - 1]) % 1000000007
            elif i == k - 1:
                a[k - 1][j] = (s[j] - a[k - 2][j - 1]) % 1000000007
            else:
                a[i][j] = (s[j] - a[i - 1][j - 1] - a[i + 1][j - 1]) % 1000000007
            s[j + 1] = (s[j + 1] + a[i][j]) % 1000000007
    print((s[l] - a[0][l - 1]) % 1000000007)

• 经验教训
  1、最开始的思考方法是使用错拍公式和排列组合的方法直接计算求值，结果发现式子没有普遍可以递归的效果。后结合参考的DP求解。
  2、注意到进制不是变量，而是一个固定值，因此二维DP中的参数为
  a[x][y]表示最高位为x，剩余y位满足K好数的数目。原理为
• S[i] = sum(a[j][i - 1])：最高位取不同值的所有结果之和
• a[i][j - 1] = S[j] - a[i-1][j-2] - a[i+1][j-2]:最高位为i的数字的K好数相当于次高位不为i-1或i+1的情况。
  3、第一次有得分的版本为（90分）COU运行时间为78ms

k, l = list(map(int, input().split()))
s = [0 for i in range(l + 1)]
a = [[0 for i in range(l)] for i in range(k)]
if l == 1:
    print(k)
else:
    for i in range(k):
        a[i][0] = 1
    s[1] = k 
    for  j in range(1,l):
        for i in range(k):
            if i == 0:
                a[0][j] = s[j] - a[1][j - 1]
            elif i == k - 1:
                a[k - 1][j] = s[j] - a[k - 2][j - 1]
            else:
                a[i][j] = s[j] - a[i - 1][j - 1] - a[i + 1][j - 1]
            s[j + 1] = s[j + 1] + a[i][j]
    print((s[l] -a[0][l - 1]) % 1000000007)

之后在循环节内添加取模用于加速加法运算，改进代码为

# K好数
k, l = list(map(int, input().split()))
s = [0 for i in range(l + 1)]
a = [[0 for i in range(l)] for i in range(k)]
# a[x][y]表示最高位为x，剩余长度为y的K好数数目
if l == 1:
    print(k)
else:
    for i in range(k):
        a[i][0] = 1
    s[1] = k 
    for  j in range(1,l):
        for i in range(k):
            if i == 0:
                a[0][j] = (s[j] - a[1][j - 1]) % 1000000007
            elif i == k - 1:
                a[k - 1][j] = (s[j] - a[k - 2][j - 1]) % 1000000007
            else:
                a[i][j] = (s[j] - a[i - 1][j - 1] - a[i + 1][j - 1]) % 1000000007
            s[j + 1] = (s[j + 1] + a[i][j]) % 1000000007
    print((s[l] - a[0][l - 1]) % 1000000007)

此版本的CPU运行时间更新为15ms。但是得分依然是90，找不出问题所在，但是可以预估到应该在边界出现的问题。正好蓝桥官网允许第一个测评点下载，发现问题出在(1, 1)输入上，检查更新

k, l = list(map(int, input().split()))
s = [0 for i in range(l + 1)]
a = [[0 for i in range(l)] for i in range(k)]
# a[x][y]表示最高位为x，剩余长度为y的K好数数目
if l == 1:
    if k == 1:
        print(0)
    else:
        print(k)
else:
    for i in range(k):
        a[i][0] = 1
    s[1] = k 
    for  j in range(1,l):
        for i in range(k):
            if i == 0:
                a[0][j] = (s[j] - a[1][j - 1]) % 1000000007
            elif i == k - 1:
                a[k - 1][j] = (s[j] - a[k - 2][j - 1]) % 1000000007
            else:
                a[i][j] = (s[j] - a[i - 1][j - 1] - a[i + 1][j - 1]) % 1000000007
            s[j + 1] = (s[j + 1] + a[i][j]) % 1000000007
    print((s[l] - a[0][l - 1]) % 1000000007)

OJ要求(1, 1)输入时输出应当为0。。不知道为什么，耍了点小聪明把分数拿到手了。