算法训练 K好数

122 阅读1分钟

小知识,大挑战!本文正在参与“程序员必备小知识”创作活动。

问题描述

如果一个自然数N的K进制表示中任意的相邻的两位都不是相邻的数字,那么我们就说这个数是K好数。求L位K进制数中K好数的数目。例如K =4,L = 2的时候,所有K好数为11、13、20、22、30、31、33共7个。由于这个数目很大,请你输出它对1000000007取模后的值。

输入格式

输入包含两个正整数,K和L。

输出格式

输出一个整数,表示答案对1000000007取模后的值。

样例输入

42

样例输出

7

数据规模与约定

对于30%的数据,KL <=106;

对于50%的数据,K<= 16, L <= 10;

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

参考代码

import java.util.Scanner;

public class Main {

	/**
	 * @param args
	 */
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		Scanner sc=new Scanner(System.in);
		int K=sc.nextInt();
		int L=sc.nextInt();
		int num[][]=new int[L][K];
		for(int i=0;i<K;i++){
			num[0][i]=1;
			
		}
		for(int i=1;i<L;i++){
			for(int j=0;j<K;j++){
				for(int x=0;x<K;x++){
					if(x!=j-1 && x!=j+1){
						num[i][j]+=num[i-1][x];
						 num[i][j]%=1000000007;
					}
				}
			}
		}
		
		int result=0;
		for(int i=1;i<K;i++){
			result+=num[L-1][i];
			result%=1000000007;
		}
		System.out.println(result);
	}

}