Leetcode 576. 出界的路径数

给定一个 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. 球一旦出界，就不能再被移动回网格内。
2. 网格的长度和高度在 [1,50] 的范围内。
3. N 在 [0,50] 的范围内。

 

这道题目使用记忆话搜索加dp转移方程解决,用一个三维状态dp[i][j][k]表示位置在k点的球，移动N次，出界的方案数。

转移方程dp[i][j][k] = dp[i+dx][j+dy][k-1] if 球没有出界，如果球出界，dp[i][j][k]+=1

这种转移方程不能直接通过循环递推得到，需要使用记忆话搜索的方法

* [576] 出界的路径数
 * dp[i][j][k] 表示在i,j,最多可以移动k次球出界的路径数
 *  
 */
class Solution {
    int[] dx={-1,0,1,0},dy={0,1,0,-1};
    int mod = 1000000007;
    public int helper(int[][][] dp,int m,int n,int k,int x,int y){
        // 这个函数的作用是求解dp[x][y][k]
        if(dp[x][y][k]!=-1) return dp[x][y][k];
        dp[x][y][k] = 0;
        if(k==0) return dp[x][y][k];
        for(int i=0;i<4;i++){
            int a = x+dx[i], b = y+dy[i];
            if(a<0 ||a==m || b<0 || b==n) dp[x][y][k]++; // 出界
            else dp[x][y][k]+=helper(dp,m,n,k-1,a,b);
            dp[x][y][k]%=mod;
        }
        return dp[x][y][k];
    }
    public int findPaths(int m, int n, int N, int x, int y) {
        int[][][] dp = new int[m][n][N+1];
        for(int i=0;i<m;i++){
            for(int j=0;j<n;j++){
                for(int k=0;k<=N;k++){
                    dp[i][j][k] = -1;
                }
            }
        }
        return helper(dp,m,n,N,x,y);
    }
}