Leetcode---329. 矩阵中的最长递增路径---每日一题---优先队列

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;
    }
};

加粗样式

你可能感兴趣的:(优先队列,Leetcode,每日一练)