k好数（Java）

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 <= 10^6；

对于50%的数据，K <= 16， L <= 10；

对于100%的数据，1 <= K,L <= 100。

---

分析

动态规划

以k=4，l=3试列出所有的k好数

发现，到后面都是在做重复的工作。
试着从这个生成的过程的最后一步来思考：

于是就找到了递推式：
dp[i][j] += dp[i-1][k]
其中，i表示个、十、百，j、k表示该位置填的数[0,n-1]

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

public class Main {

	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc = new Scanner(System.in);
		final int mod = 1000000007;
		int k,l;
		k = sc.nextInt();
		l = sc.nextInt();
		if(k == 1 && l == 1)//不知道为啥1位一进制是0而不是1
			System.out.println(0);
		else{
			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 m = 0; m < k; m++) {
						if(Math.abs(m - j) != 1) {
							if((i-1) == 0 && m == 0)
								continue;
							dp[i][j] += dp[i-1][m];
							dp[i][j] %= mod;
						}
					}
				}
			}
			
			long sum = 0;
			for(int i = 0; i < k; i++) {
				sum += dp[l-1][i];
				sum %= mod;
			}
			System.out.println(sum);
		}
		
	}

}

参考资料：
算法训练 K好数 （详解：题目理解+解题思路）