蓝桥杯 算法训练 ALGO-3 K好数 动态规划

算法训练 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(￣▽￣)ｄ )，循环比较判断是否符合K好数，这里要注意首位不能为0，所以无需与f[0][0]比较。每符合一次K好数条件，该位的值要+1。
　　第三位：循环同第二位，值得注意的是，我们要与前两位比较。这里有个方法，就是如果前两位本身就不满足K好数，我们无需比较，所以实际上我们直接与前二位K好数的第二位进行比较就可以。所以在k循环判断时，如果满足条件，我们不是+1，而是加上第二位满足K好数条件位置上的值即可。还有一个区别，就是第二位不需要和第一位的0比较，但是第三位要和第二位上的0进行比较。
　　第i位同理……
　　最后分析出的关系式为
$$\begin{gathered} f[0][i]=\{\begin{array}{cc}0,& 0\\ 1,& othervalues\end{array} \\ f[i][j]=f[i][j]+f[i-1][k] \end{gathered}$$
原理图：

代码如下：

//求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;
}