题目:
骑士按中国象棋的日字型移动
这一次,我们将 “骑士” 放在电话拨号盘的任意数字键(如上图所示)上,接下来,骑士将会跳 N-1 步。每一步必须是从一个数字键跳到另一个数字键。
每当它落在一个键上(包括骑士的初始位置),都会拨出键所对应的数字,总共按下 N 位数字。
你能用这种方式拨出多少个不同的号码?
因为答案可能很大,所以输出答案模 10^9 + 7。
思路:
每个位置跳到下一个位置的方位是固定的,所以只需要统计每个位置的数量即可
然后加起来
tips:
创建销毁数组时间耗费较大,故直接创建二维数组
class Solution {
private int[][] path = {{4, 6}, {6, 8}, {7, 9}, {4, 8}, {3, 9, 0}, {}, {1, 7, 0}, {2, 6}, {1, 3}, {4, 2}};
private static final int MOD = 1000000007;
// 创建和销毁要消耗大量时间,所以一次性创建一个二维数组
public int knightDialer(int N) {
int[][] dp = new int[N][10];
for (int num = 0; num < 10; num++)
dp[0][num] = 1; //第0次(跳到)num的不同号码个数为1
for (int time = 1; time < N; time++) {
for (int num = 0; num < 10; num++){
for (int arrive : path[num])
dp[time][num] = (dp[time][num] + dp[time - 1][arrive]) % MOD;
}
}
int res = 0;
for (int num = 0; num < 10; num++)
res = (res + dp[N - 1][num]) % MOD;
return res;
}
}