题目描述
如果一个自然数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
话都在酒里了,注释都在代码里了。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
//dp[i][j]表示i位数,最高位为j的有几种
//处理的过程中把最高位为0的情况也算上
//最后统计的时候不计算dp[l][0]就行
ll dp[105][105];
int l, k;
int main(){
scanf("%d%d", &k, &l);
for (int j=0; j<k; j++){
dp[1][j]=1;
}
for (int i=2; i<=l; i++){ //几位数
for (int j=0; j<k; j++){ //最高位为j
for (int cur=0; cur<k; cur++){
if (cur==j+1 || cur==j-1){
continue;
}
dp[i][j]+=dp[i-1][cur];
dp[i][j]%=1000000007;
}
}
}
int ans=0;
for (int j=1; j<k; j++){
ans+=dp[l][j];
ans%=1000000007;
}
printf("%lld", ans);
return 0;
}