题目描述
国际象棋中的骑士可以按下图所示进行移动:
.
这一次,我们将 “骑士” 放在电话拨号盘的任意数字键(如上图所示)上,接下来,骑士将会跳 N-1 步。每一步必须是从一个数字键跳到另一个数字键。
每当它落在一个键上(包括骑士的初始位置),都会拨出键所对应的数字,总共按下 N 位数字。
你能用这种方式拨出多少个不同的号码?
因为答案可能很大,所以输出答案模 10^9 + 7。
示例 1:
输入:1
输出:10
示例 2:
输入:2
输出:20
示例 3:
输入:3
输出:46
提示:
1 <= N <= 5000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/knight-dialer
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
无奈本人菜鸡,dp没有看懂。。。。分享一个超时的dfs
class Solution {
public:
int ans,mod;
vector<vector<int>> vis;
void dfs(int cur,int N,int i,int j){
if(cur>=N){
ans = (ans+1)%mod;
return;
}
int a,b;
for(int k=1;k<=8;++k){
if(k==1){
a=1;b=2;
}else if(k%2==0) swap(a,b);
else a=-a;
int ti=i+a,tj=j+b;
if(ti>=0&&ti<4&&tj>=0&&tj<3&&vis[ti][tj]==1) dfs(cur+1,N,ti,tj);
}
}
int knightDialer(int N) {
ans = 0;mod=1e9+7;
vis = vector<vector<int>>(4,vector<int>(3,1));
vis[3][0]=vis[3][2]=0;
for(int i=0;i<4;++i){
for(int j=0;j<3;++j){
if(vis[i][j] == 0) continue;
//可能跳的位置
dfs(1,N,i,j);
}
}
return ans;
}
};