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] 的范围内.

class Solution {
public:
    int cn=1e9+7;
    vector<vector<vector<int>>>f;
    int dp(int m, int n, int k, int x, int y){
        int &v=f[x][y][k];
        if(v!=-1)return v;
        v=0;
        if(!k)return v;
        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)v++;
            else v+=dp(m,n,k-1,a,b);
            v%=cn;
        }
       
        return v;
    }
    int dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};
    int findPaths(int m, int n, int N, int i, int j) {
        f=vector<vector<vector<int>>>(m,vector<vector<int>>(n,vector<int>(N+1,-1)));
        return dp(m,n,N,i,j);
    }
};