最大堆使用（leedcode378）

最新推荐文章于 2024-12-28 17:02:30 发布

原创最新推荐文章于 2024-12-28 17:02:30 发布 · 130 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#最大堆 #C++ #LeedCode

笔记专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一种使用最大堆解决LeetCode第378题“有序矩阵中的第K小元素”的方法。通过遍历矩阵并将元素按顺序加入最大堆中，保持堆大小为K，最终堆顶即为第K小的元素。

leedcode378：https://leetcode-cn.com/problems/kth-smallest-element-in-a-sorted-matrix/

最大堆法，C++代码：

class Solution {
public:
    int kthSmallest(vector<vector<int>>& matrix, int k) {
        priority_queue<int> maxHeap;
        for(auto vec:matrix){
            for(auto num:vec){
                if(maxHeap.size()<k){
                    maxHeap.push(num);
                }
                else if(maxHeap.top()>num){
                    maxHeap.pop();
                    maxHeap.push(num);
                }
            }
        }
        return maxHeap.top();
    }
};