LeetCode[215] Kth Largest Element in an Array

最新推荐文章于 2025-08-19 21:22:03 发布

海里的小稻草

最新推荐文章于 2025-08-19 21:22:03 发布

阅读量399

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode priority_queue

本文链接：https://blog.youkuaiyun.com/wrsophia_977/article/details/52716795

本文介绍了一种使用优先级队列解决寻找未排序数组中第K大元素问题的方法。通过定义一个小端优先级队列来维护当前遇到的前K个最大元素，并详细解释了遍历数组、更新队列的过程。

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

Find the kth largest element in an unsorted array. Note that it is the kth largest element in the sorted order, not the kth distinct element.

For example,
Given [3,2,1,5,6,4] and k = 2, return 5.

Note:
You may assume k is always valid, 1 ≤ k ≤ array's length.

用priority_queue，定义小端队列，priority_queue<int, vector<int>, greater<int>>，小的元素先出列，队头为最小元素，也即第K大元素

遍历nums数组，当队列大小为k的时候如果i>topK.top()，队列弹出队头，把i放进去，否则跳过元素i

class Solution {
public:
	int findKthLargest(vector<int>& nums, int k) {
		priority_queue<int, vector<int>, greater<int>> topK;
		for (auto i : nums)
		{
			if (topK.size() == k)
			{
				if (topK.top() >= i)
					continue;
				topK.pop();
			}
			topK.push(i);
		}
		return topK.top();
	}
};