/*数组中的第 K 个最大元素: 找到未排序整数数组中的第 k 个最大元素。例如,给定数组 [3,2,1,5,6,4] 和 k = 2,返回 5。*/

文章介绍了如何使用C++实现大根堆和小根堆算法,分别用于查找给定整数向量中的第k大和第k小元素,展示了优先队列在解决这类问题中的应用。

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

#include<iostream>
#include<stack>
#include <vector>
#include<string>
#include<algorithm>
#include<unordered_map>
#include <climits>
#include<queue>
#include<unordered_set>
#include<cctype>
#include <sstream>
#include<set>
using namespace std;
#define maxsize 100

//大根堆
int find(vector<int> nu,int k) {
    priority_queue<int, vector<int>> q;
    for (auto a : nu) {
        q.push(a);
    }
    for (int i = 0; i < k-1; i++) {
        q.pop();
    }
    return q.top();
}
int main() {
    vector<int> nu = { 3,2,1,5,6,4 };
    int k = 2;
    cout << find(nu, k);
    return 0;
}

//小根堆

int findKthLargest(vector<int>& nums, int k) {
    priority_queue<int, vector<int>, greater<int>> minHeap;

    // 将前 k 个元素插入小根堆
    for (int i = 0; i < k; ++i) {
        minHeap.push(nums[i]);
    }

    // 从第 k+1 个元素开始,与堆顶比较,若大于堆顶,则弹出堆顶,插入当前元素
    for (int i = k; i < nums.size(); ++i) {
        if (nums[i] > minHeap.top()) {
            minHeap.pop();
            minHeap.push(nums[i]);
        }
    }

    // 返回堆顶元素即为第 k 大的元素
    return minHeap.top();
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值