问题
如果一个自然数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
思路
题目提示使用dp,之前一直理解错了题目,最后才知道题目中的示例意思为4进制的100以内相邻的两位都不是相邻的数字有题目中的7个。
可以先设当有一位时,K进制的小于K的每一个位置都为1,例如:4进制一位时有0,1,2,3,如下表
可以认为有一位时,0~k-1进制的每一位上都为1。例:k=4,L=1时
L\K | 0 | 1 | 2 | 3 |
---|---|---|---|---|
1 | 1 | 1 | 1 | 1 |
再由一位推出两位的情况,例如4进制2位:若固定1开头则符合条件数字有11、13,为了使前后数字不相邻得到
当有两位时,K位置的值 等于上一层的本身加上不相邻之和。例:k=4,L=2时
即K0=K0+K2+K3=3
K1=K1+K3=2
K2=K0+K2=2
…
L\K | 0 | 1 | 2 | 3 |
---|---|---|---|---|
1 | 1 | 1 | 1 | 1 |
2 | 3 | 2 | 2 | 3 |
当L=3或者更大数字时可按照此方法继续推。
省去0是因为若固定0开头则不符合位数,即01不能算两位数
下面给出代码:
#include <iostream>
#include <string.h>
using namespace std;
typedef long long ll;
const int mod=1000000007;
ll dp[105][105];
int main(){
int m,n,ans=0;
cin>>m>>n;
memset(dp,0,sizeof(dp));
for(int i=0;i<m;i++){
dp[1][i]=1;
}
for(int i=2;i<=n;i++){
for(int j=0;j<m;j++){
for(int k=0;k<m;k++){
if(k==j||((k-1!=j)&&(k+1)!=j)){
dp[i][j]=(dp[i][j]+dp[i-1][k])%mod;
}
}
}
}
for(int i=1;i<m;i++){
ans=(ans+dp[n][i])%mod;
}
cout<<ans<<endl;
return 0;
}
此题让我更加理解动态规划由大问题转换小问题解决,谢谢大佬们的指正,努力