蓝桥杯 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。

解题思路

dp[i][j]表示以i为最高位的j位数 k好数数的最大值 

AC代码

#include <iostream>
#include <cstring> 
#include <cmath>
using namespace std; 
const int yu = 1000000007; 
int dp[105][105];		//dp[i][j]表示以以i为最高数的j位数 最多k好数 
int main()
{
	int k,l;
	cin >> k >> l;
	
	if(k == 1)
		cout << "0";
	else if(l == 1)
		cout << k-1 << endl;
	else {
		memset(dp,0,sizeof dp);
		for(int i=0;i<k;i++) 
			dp[i][1] = 1; 	//以i为个位的1位数（因为不是第l位，所以可以为0）  
		
		for(int j=2;j<=l-1;j++) {
			for(int i=0;i<=k-1;i++) {
				for(int t=0;t<=k-1;t++) {
					if(abs(i-t) == 1)	//相邻 
						continue;
					dp[i][j] = (dp[i][j] + dp[t][j-1])%yu;
				}		
			}
		}
		int sum=0;
		for(int i=1;i<=k-1;i++) {	//最高位不能为0 
			for(int t=0;t<=k-1;t++) {
				if(abs(i-t) == 1)
					continue;
				sum = (sum+dp[t][l-1])%yu;
			}
		}
		cout << sum << endl;			
	} 
}