Leetcode 329. 矩阵中的最长递增路径 记忆化搜索

如果每一个点dfs,找最大值,这样存在大量的重复搜索。避免重复搜索的办法是用记忆话搜索,用f[i][j]表示从i,j出发的最长路径。很容易推导出转移方程

const int dx[] = {1,0,-1,0};
const int dy[] = {0,-1,0,1};
class Solution {
public:
    int longestIncreasingPath(vector<vector<int>>& matrix) {
        int n = matrix.size();
        if(n==0) return 0;
        int m = matrix[0].size();
        vector<vector<int>> f(n,vector<int>(m,-1));
        int res = 0;
        for(int i=0;i<n;i++){
            for(int j=0;j<m;j++){
                res = max(res,dfs(matrix,f,i,j));
            }
        }
        return res;
    }

    int dfs(vector<vector<int>>& matrix, vector<vector<int>> &f, int row, int col){
        if(f[row][col]!=-1) return f[row][col];
        f[row][col] = 1;
        for(int i=0;i<4;i++){
            int x = row+dx[i];
            int y = col+dy[i];
            if(x>=0&&x<matrix.size()&&y>=0&&y<matrix[0].size()&&matrix[x][y]>matrix[row][col]){
                f[row][col] = max(f[row][col],dfs(matrix,f,x,y)+1);
            }
        }
        return f[row][col];
    }
};

猜你喜欢

转载自blog.csdn.net/wwxy1995/article/details/108466766