题目 329. 矩阵中的最长递增路径
难度 困难
给定一个整数矩阵,找出最长递增路径的长度。
对于每个单元格,你可以往上,下,左,右四个方向移动。 你不能在对角线方向上移动或移动到边界外(即不允许环绕)。
示例1:
输入: nums =
[
[9,9,4],
[6,6,8],
[2,1,1]
]
输出: 4
解释: 最长递增路径为 [1, 2, 6, 9]。
示例2:
输入: nums =
[
[3,4,5],
[3,2,6],
[2,2,1]
]
输出: 4
解释: 最长递增路径是 [3, 4, 5, 6]。注意不允许在对角线方向上移动。
解题思路
这道题目跟深度优先搜索的意思差不多,就是不断寻找,找到最长的递增路径,朴素的 dfs 会引起超时,所以多加了一个小优化,也就是采用了动态规划。
1、dp数组记录x,y这个点的最长路径,减少重复的dfs。
2、使用dfs去计算最长路径
数据结构
对于 dfs 算法,二维数组足以满足需求
算法
遍历整个矩阵,找到每个点的最长路径,并记录下来,比较 max 最大值即可。
代码(java)
class Solution {
static int[][] dir = { { 1, 0 }, { 0, -1 }, { 0, 1 }, { -1, 0 } };
public int longestIncreasingPath(int[][] matrix) {
// 越界判断
if(matrix.length==0){
return 0;
}
// 记忆化数组
int[][] dp = new int[matrix.length][matrix[0].length];
// 记录最长递增路径的长度
int maxIncRoad = 0;
for (int i = 0; i < matrix.length; i++) {
for (int j = 0; j < matrix[0].length; j++) {
/*
小优化,访问过了肯定就不用再比较了。
被访问的两种情况:
1、是之前maxIncRoad包含的一个点
2、比之前maxIncRoad还短的路,都比不过maxIncRoad
*/
if(dp[i][j]==0){
int road = dfs(i,j,dp,matrix);
if (road>maxIncRoad) {
maxIncRoad = road;
}
}
}
}
return maxIncRoad;
}
private int dfs(int x, int y, int[][] dp, int[][] grid) {
// 记录过就直接返回
if (dp[x][y]!=0) {
return dp[x][y];
}
// 每个点最少都是1个长度,所以初始化为1
dp[x][y] = 1;
// 四个方向
for (int i = 0; i < dir.length; i++) {
int row = x+dir[i][0];
int col = y+dir[i][1];
// 越界就下一个
if (row<0||row>grid.length-1||col<0||col>grid[0].length-1) {
continue;
}
// 比当前点x,y还矮,也看下一个
if (grid[row][col]<=grid[x][y]) {
continue;
}
// 记录最长的路的长度
dp[x][y] = Math.max(dp[x][y], dfs(row, col, dp, grid)+1);
}
return dp[x][y];
}
}