试题 算法训练 K好数

问题描述

如果一个自然数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。

题解

#include<iostream>
using namespace std;
int dp[105][105]; 
int main()
{
	int k,l;//k进制数，l位
	cin>>k>>l;
	for(int j=0;j<k;j++)
	{
		dp[0][j]=1;
	} 
	for(int i=1;i<l;i++)
     for(int j=0;j<k;j++)
       for(int t=0;t<k;t++)
       {
       	  if(t!=j-1&&t!=j+1)
       	     dp[i][j]+=dp[i-1][t];
       	     dp[i][j]%=1000000007;
	   }
	int sum=0;
	for(int j=1;j<k;j++)
	{
		sum+=dp[l-1][j];	
		sum%=1000000007;
	}
	cout<<sum<<endl;
}

 对于动态规划，现在还基本不太会找规律，不太会推表达式

 看了下面两篇博客，学会了些思路：

蓝桥杯，k好数，动态规划 - Marvin-wen - 博客园蓝桥杯之K好数问题_Walk the horizon-CSDN博客

首先以此样例为例来分析一下用DP做的过程：

L位K进制数，各位上数字的范围为[0,k-1]，位数范围为[1,L]

以2位四进制数为例，列出如下表格：
 

	0	1	2	3
0	1	1	1	1
1	3	2	2	3

 首先第一位0到3都可以取，因此都为1。下面分析第二位：

 若第一位为0，第二位可以取0、2、3，有3种情况，dp[1][0]=dp[0][0]+dp[0][2]+dp[0]   [3]=1+1+1=3；

 若第一位为1，第二位可以取1、3，有2种情况，dp[1][1]=dp[0][0]+dp[0][3]=1+1=2；

 若第一位为2，第二位可以取0、2，有2种情况，dp[1][2]=dp[0][0]+dp[0][2]=1+1=2；

 若第一位为3，第二位可以取0、1、3，有3种情况，dp[1][3]=dp[0][0]+dp[0][1]+dp[0]   [3]=1+1+1=3；

 由于第一位不能为0，因此最后可取总数为dp[1][1]+dp[1][2]+dp[1][3]，为7。