算法训练 K好数 蓝桥杯

问题描述

如果一个自然数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。

本身想用搜索，然后数据太大了，看了提示，要用动态规划。

也不能算典型的动态规划，因为不存在最优子结构，只是用了重复利用子问题的思想。

具体思路是建立一张表格，i表示最后一位数字是i，j表示j位数，先填j=1的那一列，然后j=2的每个元素等于j=1的元素累加，再除去最后两位相邻的情况，以此类推。

由于只用前面一列的数据，所以设立滚动数组，只需要2*K的空间。

相对于搜索，需要K的L次幂时间，用动态规划只需要解决K*L个子问题所需的时间。

#include<stdio.h>
#define min(a,b) a<b?a:b
int main(){
int i,j,k,e;
long long K,L,q=0;
long long a[1500][2]={0};
scanf("%lld%lld",&K,&L);
e=0;
for(i=0;i<K;i++)
    a[i][e]=1;
for(j=2;j<=L;j++){
    e=1-e;
    for(i=0;i<K;i++){
        a[i][e]=0;
    }
    for(i=0;i<K;i++){
       for(k=0;k<K;k++){
        if(k==i-1||k==i+1)continue;
        a[i][e]+=a[k][1-e];
        a[i][e]%=1000000007;
       }
    }
}
for(i=1;i<K;i++){
   q+=a[i][e];
   q%=1000000007;
}
printf("%lld",q);
return 0;
}