动态规划之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;
}