378. Kth Smallest Element in a Sorted Matrix

最新推荐文章于 2020-07-02 20:09:32 发布

原创最新推荐文章于 2020-07-02 20:09:32 发布 · 122 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

19 篇文章

订阅专栏

本文介绍了一种使用优先队列解决二维有序数组中查找第K大元素的方法。通过维护一个大小为K的最大堆，可以有效地找到指定位置的元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题链接

Given a n x n matrix where each of the rows and columns are sorted in ascending order, find the kth smallest element in the matrix.

Note that it is the kth smallest element in the sorted order, not the kth distinct element.

Example:

matrix = [
[ 1, 5, 9],
[10, 11, 13],
[12, 13, 15]
],
k = 8,

return 13.

Note:

You may assume k is always valid, 1 ≤ k ≤ n2.

翻译：从一个每行有序的二维数组中找到第k大的值。

思路：优先队列维护前k大。

AC代码：

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        priority_queue<int> que;
        for(int i=0;i<matrix.size();i++)
        for(int j=0;j<matrix[i].size();j++)
            que.push(matrix[i][j]);
        while(que.size()>k)
            que.pop();
        return que.top();
    }
};