k好数

  算法训练 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 <= 106；
对于50%的数据，K <= 16， L <= 10；

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

思路：二维数组第一个下标表示在给第几个位置选数，第二个下标表示给这个位置选哪个数字。

从个位开始看，个位选0 -- k - 1每个数字的情况都是1种，故dp[[1][0 -- k - 1]初始化为1。

然后依次看直到最高位，对每个位置计算选0 -- k - 1时候总共有多少种情况。

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<math.h>

int dp[105][105];

int main(){
	int k, l;
	int i, j, m, ans;
	
	scanf("%d %d", &k, &l);
	memset(dp, 0, sizeof(dp));
	
	for(i = 0; i < k; i++)
		dp[1][i] = 1;//初始化第一个位置选0 -- k - 1的情况均为1种
	
	for(i = 2; i <= l; i++){
		for(j = 0; j < k; j++){
			for(m = 0; m < k; m++){
				if(abs(m - j) != 1){
					dp[i][j] = (dp[i][j] + dp[i - 1][m]) % 1000000007;
				}
			}
		}
	}
	
	ans = 0;
	for(i = 1; i < k; i++)//最高位不能选0，故结果从选1开始加
		ans = (ans + dp[l][i]) % 1000000007;
	printf("%d\n", ans);
	return 0;
}