给定一个 m × n 的网格和一个球。球的起始坐标为 (i,j) ,你可以将球移到相邻的单元格内,或者往上、下、左、右四个方向上移动使球穿过网格边界。但是,你最多可以移动 N 次。找出可以将球移出边界的路径数量。答案可能非常大,返回 结果 mod 109 + 7 的值。
示例 1:
输入: m = 2, n = 2, N = 2, i = 0, j = 0
输出: 6
解释:
示例 2:
输入: m = 1, n = 3, N = 3, i = 0, j = 1
输出: 12
解释:
说明:
球一旦出界,就不能再被移动回网格内。
网格的长度和高度在 [1,50] 的范围内。
N 在 [0,50] 的范围内。
思路:典型的记忆化搜索呀,定义dp[i][j][k]:表示走到(i,j)并且还能再走最多k步的方案数,如果当前状态遍历过直接返回答案即可。
class Solution {
private int[][][] dp;
private int mod=1000000007;
public int findPaths(int m, int n, int N, int i, int j) {
dp=new int[m][n][N+1];
for(int ii=0;ii<m;ii++)
for(int jj=0;jj<n;jj++)
for(int k=0;k<=N;k++)
dp[ii][jj][k]=-1;
return dfs(i,j,m,n,N);
}
private int dfs(int x,int y,int m,int n,int num) {
if(num<0) return 0;
if(x<0 || x>=m || y<0 || y>=n) return 1;
if(dp[x][y][num]!=-1) return dp[x][y][num];
int res=0;
res=(res+dfs(x-1,y,m,n,num-1))%mod;
res=(res+dfs(x+1,y,m,n,num-1))%mod;
res=(res+dfs(x,y-1,m,n,num-1))%mod;
res=(res+dfs(x,y+1,m,n,num-1))%mod;
return dp[x][y][num]=res;
}
}