蓝桥杯 K好数Java实现

问题描述
如果一个自然数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。
直接上AC代码,参考了博主
百慕大
https://blog.csdn.net/qq_29820901/article/details/78386519
加上一些自己的理解
import java.util.Arrays; import java.util.Scanner; public class Main { private static final int num = 1000000007; public static void main(String[] args) { Scanner scanner = new Scanner(System.in); int K = scanner.nextInt(); int L = scanner.nextInt(); // int 不够,并且每一步求余，防止局部过大 // 测试最大100 long[][] dp = new long[101][101]; // 一位肯定是好数,第一行给满1个 Arrays.fill(dp[1], 1); // 从二位开始 for (int i = 2; i <= L; i++) { // 规律，从2位开始遍历，下一位的数等于下一个2位加上上一个2位的和,比如求3位3进制，先求得2位11,20,22,3个再求下一个2位11,20,22,3个 // 组成111,222,220,202,200,5个,因为下一个2位中20接不上，12接不上21接不上，结果为2 // 开头不以0如01,02....但是中间量有0，j从0开始,而最高位从L开始 int j = 0; if (i == L) { j = 1; } for (; j < K; j++) { for (int k = 0; k < K; k++) { if (Math.abs(k - j) != 1) { // 加上上一位同数字开头所有情况,因为同数字肯定符合，不同数字与其相同数字比较，结果肯定全部覆盖 dp[i][j] += dp[i - 1][k] % num; } } } } long sum = 0; // 把L位每一个数字开头加起来,1开始 for (int i = 1; i < K; i++) { sum += dp[L][i] % num; } System.out.println(sum % num); }}