原题地址:https://leetcode-cn.com/problems/longest-increasing-path-in-a-matrix/submissions/
题目描述:
给定一个整数矩阵,找出最长递增路径的长度。
对于每个单元格,你可以往上,下,左,右四个方向移动。 你不能在对角线方向上移动或移动到边界外(即不允许环绕)。
示例 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]。注意不允许在对角线方向上移动。
解题方案:
首先想到的是深度优先遍历算法,这种方法使用了二维数组定义四个方向。深度优先遍历,由一个节点开始一直遍历。这里的代码还未作出优化,时间复杂度比较大。
代码:
class Solution {
public:
vector> state = {{0, 1}, {1, 0}, {0, -1}, {-1, 0}};
int longestIncreasingPath(vector>& matrix) {
int n = matrix.size();
if (n == 0)
return 0;
int m = matrix[0].size();
vector> dp(n, vector(m, 0));
int res = 0;
for (int i = 0; i < n; i++)
for (int j = 0; j < m; j++)
res = max(dfs(dp, matrix, i, j), res);
return res;
}
int dfs(vector>& dp, vector> matrix, int i, int j) {
if (dp[i][j] != 0)
return dp[i][j];
dp[i][j] = 1;
for (vector s : state) {
int x = i + s[0];
int y = j + s[1];
if (x >= 0 && x < matrix.size() && y >= 0 && y < matrix[0].size() && matrix[i][j] < matrix[x][y])
dp[i][j] = max(dp[i][j], dfs(dp, matrix, x, y) + 1) ;
}
return dp[i][j];
}
};
时间复杂度最小的答案:
class Solution {
public:
int m, n;
vector> memo;
int dfs(vector>& matrix, int x, int y) {
if(memo[x][y] != -1)
return memo[x][y];
int ret = 1;
if(x>0 && matrix[x-1][y]>matrix[x][y])
ret = max(ret, 1 + dfs(matrix, x-1, y));
if(xmatrix[x][y])
ret = max(ret, 1 + dfs(matrix, x+1, y));
if(y>0 && matrix[x][y-1]>matrix[x][y])
ret = max(ret, 1 + dfs(matrix, x, y-1));
if(ymatrix[x][y])
ret = max(ret, 1 + dfs(matrix, x, y+1));
memo[x][y] = ret;
return ret;
}
int longestIncreasingPath(vector>& matrix) {
m = matrix.size();
if(m == 0) return 0;
n = matrix[0].size();
memo.resize(m);
int ans = 1;
for(int i = 0; i < m; ++i) memo[i].resize(n, -1);
for(int i = 0; i < m; ++i) {
for(int j = 0; j < n; ++j) {
int temp = dfs(matrix, i, j);
ans = ans < temp ? temp : ans;
}
}
return ans;
}
};