蓝桥杯 Python 练习题 k好数

试题 算法训练 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。

---

 动态规划：完全正确

 dp数组的列代表位数，行代表以什么数结尾

k,l=map(int,input().split())
su=0
dp=[[0 for _ in range(l)]for _ in range(k)]
for i in range(1,k):
    dp[i][0]=1
for i in range(1,l):#动态规划
    su=0
    for j in range(k):
        ways=0
        for h in range(k):
            if(j==0):#结尾是0的情况
                if(h!=1):
                    ways+=dp[h][i-1]
            elif(j==k-1):#结尾是k-1的情况
                if(h!=k-2):
                    ways+=dp[h][i-1]
            else:
                if(h!=j+1 and h!=j-1):
                    ways+=dp[h][i-1]
        su+=ways#为了求最后一列相加
        dp[j][i]=ways
print(su%1000000007)

暴力版：疯狂超时超限

import itertools
k,l=map(int,input().split())
a=[i for i in range(0,k)]
#print(a)
a=list(itertools.product(a,repeat=l))#生成笛卡尔积
#print(a)
total=0
for h in a:
    if(h[0]!=0):
        for j in range(1,l):
            if(h[j-1]==h[j]+1 or h[j-1]==h[j]-1):
                print(h[j],h[j]+1,h[j]-1)
                break
            if (j==l-1):
                print(h)
                total+=1
print(total%1000000007)

关于生成排列组合的内置函数

python中的排列组合内置函数_ws_nlp_的博客-CSDN博客_python中排列组合函数https://blog.csdn.net/qq_33624866/article/details/107555112?spm=1001.2101.3001.6650.16&utm_medium=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ERate-16.pc_relevant_default&depth_1-utm_source=distribute.pc_relevant.none-task-blog-2%7Edefault%7EBlogCommendFromBaidu%7ERate-16.pc_relevant_default&utm_relevant_index=18