关键词: 动态规划(深层次的迭代过程)
问题描述
如果一个自然数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进制中每一个数结尾的可能结果一共有多少种。接着第二位,在第一位的基础上以k进制的每一位结尾一共有多少种,一直到第L位。
例如:以题目中的例子 K=4,L=2。4进制,位数为2。在4进制下,一共有0,1,2,3这4个数字。
第一位以0结尾,有一种;以1结尾,有一种;以2结尾,有一种;以3结尾,有一种;
即0,1,2,3这4个数:
可以发现,每增加一行(增加的新的一位时),对新一行每个数遍历时:每个数都是选择上一行(前面的L-1位数)中与此数不相邻的数累加,得到的是有L位数时尾数为x的组合有多少种。因此最终还要sum累加第L行所有可能的尾数的种数。
代码实现如下:
#include<bits/stdc++.h> //万能的头文件
using namespace std;
int main(){
int K,L=0;
cin>>K>>L;
long long dp[150][150],sum=0;
//K=1时,每个数都是好数,因此全部初始化为1
for(int i=0;i<K;i++){
dp[1][i]=1;
}
//从第二行开始迭代,计算K行(也就是K位时)
for(int i=2;i<=L;i++){
for(int j=0;j<K;j++){
for(int x=0;x<K;x++){
if(x!=j-1&&x!=j+1){
dp[i][j]+=dp[i-1][x];//累积上一行不相邻的好数再进行遍历
dp[i][j]%=1000000007;
}
}
}
}
//统计第K行每个数字下的情况并相加
for(int p=1;p<K;p++){
sum+=dp[L][p];
sum%=1000000007;
}
cout<<sum<<endl;
return 0;
}