378.leetcode Kth Smallest Element in a Sorted Matrix(medium)[堆求第K小的 ]

最新推荐文章于 2020-07-19 10:21:13 发布

原创最新推荐文章于 2020-07-19 10:21:13 发布 · 288 阅读

0 ·

CC 4.0 BY-SA版权

C++学习专栏收录该内容

66 篇文章

订阅专栏

本文介绍了一种使用最大堆解决矩阵中寻找第K小元素的问题。通过遍历矩阵并将元素依次加入最大堆，始终保持堆中有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.

采用堆的思想来做，这个堆存放一个K个元素的最大堆，堆顶元素元素就是所求

class Solution {
public:
    //采用堆的思想来做，这个堆存放一个K个元素的最大堆，堆顶元素元素就是所求
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        if(matrix.size()<=0) return 0;
        priority_queue<int> ipq;
        for(int i = 0;i<matrix.size();i++)
            for(int j = 0;j<matrix[i].size();j++)
            {
                ipq.push(matrix[i][j]);
                if(ipq.size()>k)
                     ipq.pop();
            }
        return ipq.top();
    }
};