K好数【DP】

问题描述

如果一个自然数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%的数据，K^L <= 10⁶；

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

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

#include<stdio.h>
#include<string.h>
int dp[110][110];
int main()
{
	int i, j, k, n, m;
	while(scanf("%d %d",&m, &n) != EOF)
	{
		if(n == 1)
		{
			printf("%d\n", m-1);
			continue;
		}
		memset(dp, 0, sizeof(dp));
		for(i = 1; i < m; ++i)
		dp[1][i] = 1;
		int ans = 0;
		for(i = 2; i <= n; ++i)
		{
			for(j = 0; j < m; ++j)
			{
				for(k = 0; k < m; ++k)
				if( k != j-1 && k != j+1)
				{
					dp[i][j] += dp[i-1][k];
					if(dp[i][j] > 1000000007)
					dp[i][j] %= 1000000007;
				}
				if(i == n)
				{
					ans += dp[i][j];
					ans %= 1000000007;
				}
			}
		}
		printf("%d\n", ans);
	 } 
	 return 0;
 } 

思路：用一个DP数组保存长度为i末位为j时的情况数，初始化长度为1时的情况数为1，当长度大于1时，要处理好首位为0的情况，所以我们将dp[1][0]设置为0，即表示当长度为2时转移方程dp[2][j] += dp[1][0]*dp[1][j]（默认为1省略）时保证首位为0的情况不会统计进去。  当长度为2为时首位为0的未统计进去，那么长度大于2的都不会统计进去。

转移方程 dp[i][j] += dp[i-1][k]*1   （k!=j-1 && k!=j+1）。