算法训练 K好数
时间限制:1.0s 内存限制:256.0MB
问题描述
如果一个自然数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好数标准,所以二维数组该位置的值应该是1。
第二位:每一位数都要和第一位进行比较,符合K好数就+1,所以还要用一个for循环,这里用k来循环(个人喜欢i,j,k; o( ̄▽ ̄)d ),循环比较判断是否符合K好数,这里要注意首位不能为0,所以无需与f[0][0]比较。每符合一次K好数条件,该位的值要+1。
第三位:循环同第二位,值得注意的是,我们要与前两位比较。这里有个方法,就是如果前两位本身就不满足K好数,我们无需比较,所以实际上我们直接与前二位K好数的第二位进行比较就可以。所以在k循环判断时,如果满足条件,我们不是+1,而是加上第二位满足K好数条件位置上的值即可。还有一个区别,就是第二位不需要和第一位的0比较,但是第三位要和第二位上的0进行比较。
第i位同理……
最后分析出的关系式为
f
[
0
]
[
i
]
=
{
0
,
0
1
,
o
t
h
e
r
v
a
l
u
e
s
f
[
i
]
[
j
]
=
f
[
i
]
[
j
]
+
f
[
i
−
1
]
[
k
]
f[0][i]= \left \{\begin{array}{cc} 0, &0\\ 1, & other\ values \end{array}\right.\\ \\ f[i][j] = f[i][j] +f[i-1][k]
f[0][i]={0,1,0other values f[i][j]=f[i][j]+f[i−1][k]
原理图:
代码如下:
//求L位K进制数中K好数的数目。
#include <iostream>
using namespace std;
int main()
{
int K, L, **f;
int i, j, k, sum = 0;
cin >> K >> L;
//构建二位数组
f = new int*[L];
for(i = 0; i < L; i++) f[i] = new int[K];
//依题意初始化
for(i = 1; i < K; i++) f[0][i] = 1;
f[0][0] = 0;
for(i = 1; i < L; i++)
{
for(j = 0; j < K; j++)
{
f[i][j] = 0;
}
}
//计算
if(L > 1)
{
for(i = 1; i < L; i++)
{
for(j = 0; j < K; j++)
{
for(k = 0; k < K; k++)
if(k != j - 1 && k != j + 1)
{
f[i][j] += f[i-1][k];
f[i][j] %= 1000000007;
}
}
}
}
//求和
for(i = 0; i < K; i++)
{
sum += f[L-1][i];
sum %= 1000000007;
}
cout << sum << endl; //输出结果
return 0;
}