leetcode——378. 有序矩阵中第 K 小的元素

最新推荐文章于 2025-05-19 05:00:00 发布

原创最新推荐文章于 2025-05-19 05:00:00 发布 · 158 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 专栏收录该内容

145 篇文章

订阅专栏

本文探讨了两种对二维数组排序的方法。第一种是将数组转换为一维并直接排序，但未充分利用每行和列的有序性。第二种方法采用优先队列，利用每行有序的特点，通过不断压入最小值来找到第k小的元素，提高了效率。

思路1

将二维数组转为一位数组进行排序
但是这样没有利用到 每行和每列元素均按升序排序

代码1

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        vector<int> ans;
        for(int i=0;i<matrix.size();i++)
            for(int j=0;j<matrix[0].size();j++)
                ans.emplace_back(matrix[i][j]);
        sort(ans.begin(),ans.end());
        return ans[k-1];
    }
};

思路2

使用优先队列，结构为<val,x,y>
按照每行有序的特点，每次压入栈顶的右元素，保证其为较小的值；
（如果其不是当前元素的下一个值，那么就会是下一行的头元素）

代码2

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        struct point {
            int val, x, y;
            point(int val, int x, int y) : val(val), x(x), y(y) {}
            bool operator> (const point& a) const { return this->val > a.val; }
        };
        
        //创建优先队列 存入每行头地址
        priority_queue<point, vector<point>, greater<point>> que;
        int n = matrix.size();
        for (int i = 0; i < n; i++) 
            que.emplace(matrix[i][0], i, 0);

        //每次压入当前栈顶的右边元素  若当前元素是最右边元素则不压入
        for (int i = 0; i < k - 1; i++) {
            point now = que.top();
            que.pop();
            if (now.y != n - 1) 
                que.emplace(matrix[now.x][now.y + 1], now.x, now.y + 1);
        }
        return que.top().val;
    }
};