问题描述
如果一个自然数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%的数据,K^L <= 106;
对于50%的数据,K <= 16, L <= 10;
对于100%的数据,1 <= K,L <= 100。
思路: 这是一道动态规划题,找到合适的子问题对解决这道题至关重要。对于不同的K, L ,我们采用填充数位的方法。比如上次的数位上的数字为xx,这次扩充一位为xx_, 那么我们要填充的数字y, 以y结尾的K好数当然与xx的K好数有关,以此为例当数位是3,要填充y,以 y 结尾的K 好数 dp[3] [y] = dp[3][y] + dp[2][x]; 其中x != y - 1 且 x != y + 1;
这样便可找到状态转移方程 dp[i][j] = dp[i][j] + dp[i - 1][x] ; (x != j - 1 && x != j +1)
dp[i][j] 表示 以 j 结尾的有i位数的K 好数(这里是包含前导数零的).
需要注意的是这道题的结果不包含前导数零, 而在计算的时候是将前导数 0 也包含进去的。以此在最后将结果相加的时候从dp[L][j], j 从 1 开始。
比如样例中的k = 4, l = 2, K好数为(11、13、20、22、30、31、33)包含前导数零的话则还有00、 02、 03,可能你们会有疑问不是有可以以0结尾的吗 ?是的如20、30,注意一下反过来是什么02、03,那么20、30可以理解成以2、3结尾,这样在计算的时侯就可以抛开前导数零了。
#include<iostream>
#include<string.h>
#define cons 1000000007
using namespace std;
int main()
{
long long dp[104][104];
int k, l;
while(cin>>k>>l)
{
memset(dp, 0, sizeof(dp));
for(int i = 0; i < k; i++)//1位数以i结尾的情况
dp[1][i] = 1;
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)continue;
dp[i][j] = dp[i][j] + dp[i - 1][x];
dp[i][j] %= cons;
}
long long sum = 0;
for(int i = 1; i < k; i++)//需要注意的地方
{
sum += dp[l][i];
sum %= cons;
}
cout<<sum<<endl;
}
return 0;
}