蓝桥杯 算法训练 K好数（java实现）解析

资源限制
时间限制：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[l][k]存储数据，l表示位数，k表示进制。假设只有一位数，那么每个数都为k好数，所以第一行dp[l]初始化为1。

填表过程：
把j作为k好数的第一位，除开j相邻的结果，其他结果全部相加即为dp[i][j]，例：以0作第一位的k好数有00，02，03，所以dp[2][0]=3；再有以1作第一位的k好数有11，13，所以dp[2][1]=2。当k=4，l=2的填表结果如上。
这样可以总结状态转移方程为：

for(int m=0; m<k; m++) {
	if(m!=j+1 && m!=j-1) {
		dp[i][j] += dp[i-1][m];
	}
}

要注意的是k好数是不能以0作第一位的，所以最后计算答案时不将dp[i][0]计入。状态转移表计算0作第一位的情况是为了其他数的结果计算。

代码如下：

import java.util.Arrays;
import java.util.Scanner;

public class KGoodNumber {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int k = sc.nextInt();
		int l = sc.nextInt();
		sc.close();
		int[][] dp = new int[l+2][k+2];
		Arrays.fill(dp[1], 1);
		for(int i=2; i<=l; i++) {
			for(int j=0; j<k; j++) {
				for(int m=0; m<k; m++) {
					if(m!=j+1 && m!=j-1) {
						dp[i][j] += dp[i-1][m];
						dp[i][j] %= 1000000007;
					}
				}
			}
		}
		
		int count = 0;
		for(int i=1; i<k; i++) {
			count += dp[l][i];
			count %= 1000000007;
		}
		System.out.println(count);
	}
}