蓝桥杯训练题 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

解题思路：

首先我们看题能能发现，求某进制的3位k好数与其2位是有关系的，而1位则不用考虑，除0外，1到L-1都是k好数.

例如：

4进制的1位k好数：

1 ，2，3

4进制的2位k好数有

11、13、20、22、30、31、33

其3位有：

1 1 1、1 1 3、1 3 0、1 3 1 、1 3 3、

2 0 0、 2 0 2、 2 0 3、2 2 0、 2 2 2、

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

可以知道，L位数的k好数等于以[1,L-1]开头产生的k好数相加而成.

现在就是要弄清以相同数字开头的不同位k好数的联系，

看到11、13、20、22、30、31、33

1 1 1、1 1 3、1 3 0、1 3 1 、1 3 3

从2位扩展到3位也就是在2位最前面添加一位数，当新添加的数与原来最左边的数不相邻则是k好数，例如，11前加1是111，是k，13前加1是k，20前加1是120不是k，以此类推得到最后以1开头的3位的4进制k好数总数.当然，相信你也注意到了202，203，那按本题我们的解题思想，那应该存在00，03，然后往前加3或是2，形成新的k好数。是的，在组合时是有的，但是是以0开头，所以不计入其位数k好数总数的计算.

完整的“二位”k好数：00，02，03，11、13、20、22、30、31、33，但是实际是00，02是L-1位数，所以不计算.

好了，清楚这点，上表格：

下图是求4进制2位k好数

k代表此时的位数，L以某位数开头的k好数

k：从0开始，0代表一位数

L：从0开始，0代表以0开头的k好数位置

（0，0）表示以0开头的1位数的k好数个数，（1，0）表示以0开头的2位k好数个数

（0，0）到（0，3）初始化为1，因为个位数都是k好数.

设置数组dp[K][L];

计算（1，0），历遍一次（0，0）到（0，3），判断0和0，1，2，3是否能组成k好数，能，就把其个数加到dp（1，0）这里加了（0，0），（0，2），（0，3），所以是3，其他同理

最后把（1，1）加到（1，3）就是其2位k好数个数也就是7；

当要求4进制3时：

计算（2，1）用1和0，1，2，3比较只有1，3满足，也就是加（1，3）和（1，1）等于5，其他同理，最后计算得18；

主要思想就是动态规划，分解问题，想着求3位，怎么就从2位推过去，求2位，怎么从1位推过去.

最后代码：

import java.util.*;
public class k好数 {
public static void main(String[] args) {
	Scanner Lps=new Scanner(System.in);
	int k,l;
	k=Lps.nextInt();
	l=Lps.nextInt();
	int dp[][]=new int[l][k];
	for(int i=0;i<k;i++) {
		dp[0][i]=1;
	}
	for(int i=1;i<l;i++) {
		for(int j=0;j<k;j++) {
			for(int j1=0;j1<k;j1++) {
				if(Math.abs(j1-j)!=1) {
					dp[i][j]+=dp[i-1][j1];
					dp[i][j]%=1000000007;
				}
			}
		}
	}
	int sum=0;
	for(int i=1;i<k;i++) {
		sum+=dp[l-1][i];
		sum%=1000000007;
	}
	System.out.print(sum);
	Lps.close();
	
}
}