https://leetcode-cn.com/problems/knight-dialer/
国际象棋中的骑士可以按下图所示进行移动:
.
这一次,我们将 “骑士” 放在电话拨号盘的任意数字键(如上图所示)上,接下来,骑士将会跳 N-1 步。每一步必须是从一个数字键跳到另一个数字键。
每当它落在一个键上(包括骑士的初始位置),都会拨出键所对应的数字,总共按下 N
位数字。
你能用这种方式拨出多少个不同的号码?
因为答案可能很大,所以输出答案模 10^9 + 7
。
示例 1:
输入:1 输出:10
示例 2:
输入:2 输出:20
示例 3:
输入:3 输出:46
提示:
1 <= N <= 5000
存在问题 如果dp设置成为 int 那么 (dp1[3]+dp1[9]+dp1[0])%MOD;有可能超出范围
class Solution {
public:
int dp1[10],dp2;
const int MOD=1e9+7;
int knightDialer(int N) {
memset(dp1,1,sizeof(dp1));
long long sum=0;
for(int i=0; i<N; i++) {
for(int j=0; j<10; j++) {
sum=sum+dp1[j];
sum=sum%MOD;
}
dp2[1]=(dp1[8]+dp1[6];
dp2[2]=dp1[7]+dp1[9];
dp2[3]=dp1[8]+dp1[4];
dp2[4]=dp1[3]+dp1[9]+dp1[0];
dp2[5]=0;
dp2[6]=dp1[1]+dp1[7]+dp1[0];
dp2[7]=dp1[2]+dp1[6];
dp2[8]=dp1[1]+dp1[3];
dp2[9]=dp1[2]+dp1[4];
dp2[0]=dp1[6]+dp1[4];
memcpy(dp1,dp2,sizeof(dp1));
}
cout<<sum
}
};