蓝桥杯 K好数 Java 特详细解释

import java.math.BigInteger;
import java.util.Scanner;

public class Main {
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int K = sc.nextInt();
        int L = sc.nextInt();
        //最左位填了一个数，如果想让L位数位k好数，那么剩下的位置（整体）也必须是k好数，这满足动态规划的最优子问题
        int[][] dp = new int[L+1][K];//dp[i][j]表示有i个位置，其最左位填j所得到的k好数的个数   
        for(int i = 0;i<K;i++){
            dp[1][i] = 1;//显然只有一个位置时，K取多少都是只能填本身，所以 dp[1][i] = 1
        }
        for(int i = 2;i<=L;i++){
            for(int j = 0;j<K;j++){
                for(int k = 0;k<K;k++){
                    if(k != j-1 && k!= j+1){//k表示填完j后为后面的位置要填的数做排除，排除那些与最左位相邻的数`在这里插入代码片`
                    //例如L=3，K=4，那么当当L=3，K=2时，dp[3][2]表示我有3个位置，
                    //最左位填了**2**，那么剩下的整体(还有两位)去只有两个位置的dp数组中找（即dp[2][k]），
                    //所以转移方程为dp[i][j] += dp[i-1][k]，而这里的k就表示我要找不与**2**相邻的其他符合的数，
                    //因此最内侧的for循环为 for(int k = 0;k<K;k++)
                        dp[i][j] = (dp[i][j] + dp[i-1][k])%1000000007;
                    }
                }
            }
        }
        BigInteger ans = logs(K,L,dp);
        System.out.println(ans);
    }
    public static BigInteger logs(int K,int L,int[][] dp){
        BigInteger sum = new BigInteger("0");//创建一个值为0（字符串）的BigInteger
        for(int i = 1;i < K;i++) {
            sum = sum.add(new BigInteger(Integer.toString(dp[L][i])));//把dp中的值加进去（需要强转为字符串），就不用担心数溢出
        }
        return sum.mod(new BigInteger("1000000007"));//sum.mod()取余
    }
}