题意:
如果一个自然数N的K进制表示中任意的相邻的两位都不是相邻的数字,那么我们就说这个数是K好数。求L位K进制数中K好数的数目。
思路:
典型的动态规划题:
一位一位的去填充,用一个二维数组F[i][j]表示i位数以j结尾一共有多少种,接下来要放下一位的话,只要判断要放的这位数字是否是j的邻位即可。
参考:http://blog.csdn.net/ljd4305/article/details/21234723
链接:http://lx.lanqiao.cn/problem.page?gpid=T13
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn = 101;
const int mod = 1000000007;
int f[maxn][maxn]; //f[i][j]表示i位数,最后一位是j的k好数的个数,其中f[i][j] = sum(f[i-1][j]);
int main()
{
int l, k;
long long ans, n;
while(cin >> k>> l)
{
if(l == 1) //一位数的话就直接是k-1个了
{
cout << k-1 << endl;
continue;
}
memset(f,0,sizeof f);
for(int j = 0; j < k; j++) //如果是一位数,那么每个小于k的数就是1
{
f[1][j] = 1;
}
for(int i = 2; i < l; i++) //两位数以上的判断
{
for(int j = 0; j < k; j++) //在放好i-1位数后放下一个,这个数和循环中的i-1位数的最后每一位都不能相邻
{
int cnt = 0; //不断计算放一位数字
for(int r = 0; r < k; ++r)
{
if(r == j + 1 ||r == j - 1)continue; //判断要放的j是否是已放好的i-1位数的最后每一位r的邻位
cnt = (cnt + f[i-1][r]) % mod; //cnt的值即为其本身加上放好前i-1位且最后一位放的是r的个数,因为前i-1位数放好后继续放下一位,此时的下一位必须是循环的这一个
}
f[i][j] = cnt;
}
}
int sum = 0;
for(int j = 1; j < k; ++j) //两位数的判断,只要从1开始确定第一位即可
{
int cnt = 0;
for(int r = 0; r < k; ++r)
{
if(r == j + 1 ||r == j - 1)continue;
cnt = (cnt + f[l-1][r]) % mod;
}
sum = (sum + cnt)%mod;
}
cout<< sum <<endl;
}
return 0;
}