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]。注意不允许在对角线方向上移动。
优先队列
class Node {
public:
int r, c, v;
Node(int row, int col, int val) : r(row), c(col), v(val) {}
bool operator < (const Node& other) const {
return v > other.v;
}
};
class Solution {
public:
int longestIncreasingPath(vector<vector<int>>& matrix) {
if (matrix.empty()) return 0;
int n = matrix.size(), m = matrix[0].size(), ans = -1;
vector<vector<int>> memo(n, vector<int>(m, 0));
priority_queue<Node> pq;
for (int i = 0; i < n; i++)
for (int j = 0; j < m; j++)
pq.push(Node(i, j, matrix[i][j]));
while (pq.size()) {
Node x = pq.top(); pq.pop();
int r = x.r, c = x.c, v = x.v;
if (!memo[r][c]) memo[r][c] = 1;
if (c && matrix[r][c - 1] < matrix[r][c]) memo[r][c] = max(memo[r][c], memo[r][c - 1] + 1);
if (r && matrix[r - 1][c] < matrix[r][c]) memo[r][c] = max(memo[r][c], memo[r - 1][c] + 1);
if (c < m - 1 && matrix[r][c + 1] < matrix[r][c]) memo[r][c] = max(memo[r][c], memo[r][c + 1] + 1);
if (r < n - 1 && matrix[r + 1][c] < matrix[r][c]) memo[r][c] = max(memo[r][c], memo[r + 1][c] + 1);
ans = max(ans, memo[r][c]);
}
return ans;
}
};
加粗样式