数组中的第K个最大元素

题目描述

在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例1

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5

示例2

输入: [3,2,3,1,2,4,5,5,6] 和k = 4
输出: 4

思路

简单的思路是排序，取第K个元素，但是时间复杂度较高。经典的时间复杂度较低的两种解法，一种是利用快排的思想，迅速定位到第K大的数字，第二种思路是建立K个元素的小顶堆，维护这个堆，最后取出堆顶元素即为第K的的元素。

利用快排思想的实现

int _partition(vector<int>& v, int l, int r) {

	int data = v[l];

	while (l < r)
	{
		while (l < r && v[r] < data) r--;
		if (l < r)
		{
			v[l++] = v[r];
		}
		while (l < r && v[l] > data) l++;
		if (l < r)
		{
			v[r--] = v[l];
		}
	}
	v[l] = data;
	return l;
}

int findKthLargest(vector<int>& v, int l, int r, int k) {

	int index = _partition(v, l, r);

	if (index + 1 == k) 
	{
		return v[index];
	}

	if (index + 1 < k)
	{
		return findKthLargest(v, index + 1, r, k);
	}

	if (index + 1 > k)
	{
		return findKthLargest(v, l, index - 1, k);
	}

	return -1;
}

int findKthLargest(vector<int>& nums, int k) {
	
	return findKthLargest(nums, 0, nums.size() - 1, k);
}

利用小顶堆思想的实现

int findKthLargest(vector<int>& nums, int k) {

	priority_queue<int, vector<int>, greater<int>> data;

	for (int i = 0; i < nums.size(); i++)
	{
		data.push(nums[i]);
		if (data.size() > k)
		{
			data.pop();
		}
	}

	return data.top();
}