Problem Description
大家对进制一定很清楚,如果一个数字是A进制表示,则它的每一位数字小于A。现在定义一种奇异数字,假设它是A进制,它的每一位与它左边、右边的数字都不是相邻。现在问题是,求解B位A进制的所有整数中奇异数字的个数。比如A=3,B=2,奇异数字就有11 20 22 三个,所以结果是3.
Input
输入两个正整数A(1<=A<=100)和B(1<=B<=100).
Output
输出一个整数,结果对1000000007取模。
Sample Input
3 2
Sample Output
3
解题思路
数位dp;
dp[i][j]表示第i位上数字为j的方案数.
状态转移方程:
当 j != k+1 && j != k-1时, dp[i][j] += dp[i-1][k];
参考代码
#include <stdio.h>
#include <string.h>
#define MOD 1000000007
#define maxn 110
typedef __int64 ll;
ll dp[maxn][maxn];
int main()
{
memset(dp,0,sizeof(dp));
int a,b;
scanf("%d %d",&a,&b);
for (int i = 1;i < a;i++) dp[1][i] = 1;
for (int i = 2;i <= b;i++)
for (int j = 0;j < a;j++)
for (int k = 0;k < a;k++)
if (j != k+1 && j != k-1){
dp[i][j] += dp[i-1][k];
dp[i][j] %= MOD;
}
ll ans = 0;
for (int i = 0;i < a;i++)
ans = (ans+dp[b][i])%MOD;
printf("%I64d\n",ans);
return 0;
}