动态规划之K好数

动态规划之K好数

动态规划

动态规划是将待求解的问题分解为若干个子问题（阶段），按顺序求解子阶段，前一子问题的解，为后一子问题的求解提供了有用的信息。

问题描述

如果一个自然数N的K进制表示中任意的相邻的两位都不是相邻的数字，那么我们就说这个数是K好数。求L位K进制数中K好数的数目。例如K = 4，L = 2的时候，所有K好数为11、13、20、22、30、31、33 共7个。由于这个数目很大，请你输出它对1000000007取模后的值。

输入格式：输入包含两个正整数，K和L。

输入样例：4 2

输出样例:7

数据规模与约定:对于30%的数据，KL <= 106；对于50%的数据，K <= 16， L <= 10；对于100%的数据，1 <= K,L <= 100。

问题解析

问题要求解的是L位K进制数中K好数的个数,按照动态规划分析,将该问题拆解为若干个相似的子问题,只需要求解其中的第一个子问题,再通过迭代的方式就可以求解出整个问题的解。

关键步骤：

• 划分子问题
• 循环设计

划分子问题

L位K进制中K好数的个数划分为i位首数字为j的K好数的个数，其中i取值[1,L],j取值[0,K]，将整个问题分为L*(K+1)个子问题。创建一个二维数组f[i][j]表示i位首数字为j的K好数的个数，每个f[i][j]的值只和f[i-1][]的值有关。

循环设计

这里因为已经知道循环的确切的次数，所有直接用嵌套的for循环就可以实现

代码展示

#include<iostream>
using namespace std;
long long f[101][101]={0};	//f[i][j]表示长度为i，首数字为j的k好数的个数 
int main()
{
	int k,l;
	long long re=0;
	cin>>k>>l;
	//长度为1的所有数由于没有相邻数，所有都是k好数 
	for(int i=0;i<k;i++){
		f[1][i]=1;	
	}
	//从长度为2的数开始查找 
	for(int i=2;i<=l;i++){
		for(int j=0;j<k;j++){
			for(int m=0;m<k;m++){
				//当j数字和i-1位的首数字m不相邻时
				//便可以将 j数字和该i-1位的数字组成新的i位k好数
				//该i位的k好数的个数增加了该i-1位k好数的个数 
				if(m!=j-1&&m!=j+1){
					f[i][j]+=f[i-1][m];
					//当数量超过就取余 
					if(f[i][j]>=1000000007){
						f[i][j]%=1000000007;	
					} 
					
				}
			}
		}
	} 
	for(int i=1;i<k;i++){
		re+=f[l][i];
		if(re>1000000007){
			re%=1000000007;
		}
	}
	cout<<re;
	return 0;
}