出界的路径数--dfs记忆化+剪枝

Posted 2021-09-05 C_YCBX Py_YYDS

文章目录

• • 题目
  • 题目解析
  • 解题代码

题目

题目解析

一看到这路径题，很快想到dfs+记忆化，然后就会出现一个问题：在最后几个测试点卡死，我也就是被这些坑了很久，最后发现，还需要剪枝！

解题代码

class Solution {
public:
    constexpr  static  int dx[] = {-1,1,0,0};
    constexpr static    int dy[] = {0,0,-1,1};
    constexpr static     int MOD = 1e9+7;
    int findPaths(int m, int n, int maxMove, int startRow, int startColumn) {
        if(maxMove==0)
            return 0;
        memset(memo,0,sizeof memo);
        return dfs(0,startRow,startColumn,maxMove,m,n);
    }
private:
    int memo[50][50][51];
    int dfs(int k,int x,int y,int maxMove,int m,int n){
            if(x>=m||y>=n||x<0||y<0)
                return 1;
//关键剪枝，一旦这个位置前后左右移动能移动的步数也无法越出界限，则该点不可能再出界
            if( x>maxMove-k&&y>maxMove-k&&m - x > maxMove-k &&  n - y > maxMove-k) return 0;
            if(k==maxMove)
                return 0;
            if(memo[x][y][k])return memo[x][y][k];
                int res = 0;
            for(int i=0;i<4;i++){
                res = (dfs(k+1,x+dx[i],y+dy[i],maxMove,m,n)+res)%MOD;
            }
            return memo[x][y][k] = res;
        }
};