数组中的第 K 个最大元素(C++实现)

数组中的第 K 个最大元素

题目


数组中的第 K 个最大元素


思路

通过使用优先队列(最大堆)来找到数组中第k大的元素。通过弹出最大堆中的前k-1个元素,留下堆中的顶部元素作为结果返回。

代码

c 复制代码
class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        priority_queue<int> pq(nums.begin(),nums.end());
        int i=0;
        while(i<k-1)
        {
            pq.pop();
            i++;
        }
        return pq.top();
    }
};

代码讲解:


c 复制代码
priority_queue<int> pq(nums.begin(), nums.end());

在函数内部,创建了一个名为pq的优先队列(优先级队列),它是一个最大堆。通过将nums数组的元素从begin()到end()范围内添加到优先队列中,初始化了一个包含数组所有元素的最大堆。


c 复制代码
int i = 0;
        while (i < k - 1) {
            pq.pop();
            i++;
        }

接下来,使用一个循环,执行k-1次pq.pop()操作,将最大堆中的前k-1个元素弹出。由于最大堆的性质,每次弹出的都是当前堆中的最大元素。


c 复制代码
return pq.top();
    }
};

最后,返回最大堆中的顶部元素,即第k大的元素。由于最大堆的性质,堆顶元素即为堆中的最大元素。


下面是添加了注释的代码:

c 复制代码
class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        // 创建一个最大堆,用于存储数组元素
        priority_queue<int> pq(nums.begin(), nums.end());

        int i = 0;
        // 弹出最大堆中的前 k-1 个元素
        while (i < k - 1) {
            pq.pop();
            i++;
        }

        // 返回最大堆的顶部元素,即第 k 大的元素
        return pq.top();
    }
};

(本题完)

相关推荐
明月看潮生42 分钟前
编程与数学 02-017 Python 面向对象编程 23课题、测试面向对象的程序
开发语言·python·青少年编程·面向对象·编程与数学
TPBoreas1 小时前
架构设计模式七大原则
java·开发语言
Pafey2 小时前
【Deepseek】Windows MFC/Win32 常用核心 API 汇总
c++·windows·mfc
nightunderblackcat2 小时前
新手向:Python开发简易股票价格追踪器
开发语言·python
侯杰2 小时前
C++ 线程安全初始化机制详解与实践
c++
熙客2 小时前
Java:LinkedList的使用
java·开发语言
每天敲200行代码3 小时前
QT 概述(背景介绍、搭建开发环境、Qt Creator、程序、项目文件解析、编程注意事项)
c++·qt
大飞pkz3 小时前
【Lua】题目小练12
开发语言·lua·题目小练
赵得C3 小时前
Java 多线程环境下的全局变量缓存实践指南
java·开发语言·后端·spring·缓存