剑指 Offer II 112. 最长递增路径-----记忆化搜索
题目表述
给定一个 m x n 整数矩阵 matrix ,找出其中 最长递增路径 的长度。
对于每个单元格,你可以往上,下,左,右四个方向移动。 不能 在 对角线 方向上移动或移动到 边界外(即不允许环绕)。
输入:matrix = [[9,9,4],[6,6,8],[2,1,1]]
输出:4
解释:最长递增路径为 [1, 2, 6, 9]。
记忆化搜索
class Solution {
int[][] dirs = new int[][]{{1,0}, {-1, 0}, {0, -1}, {0, 1}};
int[][] dis;
int res = 0;
public int longestIncreasingPath(int[][] matrix) {
int m = matrix.length;
int n = matrix[0].length;
dis = new int[m][n];
for(int i = 0; i < m;i++){
for(int j = 0; j < n;j++){
dis[i][j] = dfs(matrix, i,j,m,n);
res = Math.max(dis[i][j], res);
}
}
return res;
}
public int dfs(int[][] matrix, int x, int y, int m, int n){
if(dis[x][y] != 0) return dis[x][y];
dis[x][y]++;
int max = 0;
for(int[] dir : dirs){
int i = x + dir[0];
int j = y + dir[1];
if(i < 0 || i >= m || j < 0 || j >=n) continue;
if(matrix[x][y] >= matrix[i][j]) continue;
dis[x][y] = Math.max(dfs(matrix, i, j, m, n) + 1, dis[x][y]);
}
return dis[x][y];
}
}