数组中的第K个最大元素--力扣

Posted 2021-07-11 穿迷彩服的鲨鱼

数组中的第K个最大元素

• 前言
• 一、示例
• 二、代码解析
• • 1.sort()库函数
  • • 代码如下（示例）：
    • 结果
  • 2.基于快速排序的选择方法
  • • 代码如下（示例）：
    • 结果
  • 测试
  • • 代码如下（示例）：
    • 结果
• 总结

---

前言

在未排序的数组中找到第 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 总是有效的，且 1 ≤ k ≤ 数组的长度。

二、代码解析

1.sort()库函数

代码如下（示例）：

//利用sort函数
int findKthLargest(vector<int>& nums, int k) 
{
    sort(nums.begin(), nums.end(), greater<int>());
    return nums[k - 1];
}

结果

2.基于快速排序的选择方法

代码如下（示例）：

/*在 c/c++ 中，为了解决一些频繁调用的小函数大量消耗栈空间（栈内存）的问题，特别的引入了 inline 修饰符，表示为内联函数*/
inline int partition(vector<int>& a, int l, int r)
{
	int x = a[r], i = l - 1;
	for (int j = l; j < r; ++j)
	{
		if (a[j] <= x)
		{
			swap(a[++i], a[j]);
		}
	}
	swap(a[i + 1], a[r]);
	return i + 1;
}

inline int randomPartition(vector<int>& a, int l, int r)
{
	int i = rand() % (r - l + 1) + l;
	swap(a[i], a[r]);
	return partition(a, l, r);
}

int quickSelect(vector<int>& a, int l, int r, int index)
{
	int q = randomPartition(a, l, r);
	if (q == index)
	{
		return a[q];
	}
	else
	{
		return q < index ? quickSelect(a, q + 1, r, index) : quickSelect(a, l, q - 1, index);
	}
}
//基于快速排序的选择方法
int findKthLargest(vector<int>& nums, int k)
{
	srand(time(0));
	return quickSelect(nums, 0, nums.size() - 1, nums.size() - k);
}

结果

测试

代码如下（示例）：

#include<vector>
#include <iostream>
using namespace std;

//利用sort函数
//int findKthLargest(vector<int>& nums, int k) 
//{
//    sort(nums.begin(), nums.end(), greater<int>());
//    return nums[k - 1];
//}

/*在 c/c++ 中，为了解决一些频繁调用的小函数大量消耗栈空间（栈内存）的问题，特别的引入了 inline 修饰符，表示为内联函数*/
inline int partition(vector<int>& a, int l, int r)
{
	int x = a[r], i = l - 1;
	for (int j = l; j < r; ++j)
	{
		if (a[j] <= x)
		{
			swap(a[++i], a[j]);
		}
	}
	swap(a[i + 1], a[r]);
	return i + 1;
}

inline int randomPartition(vector<int>& a, int l, int r)
{
	int i = rand() % (r - l + 1) + l;
	swap(a[i], a[r]);
	return partition(a, l, r);
}

int quickSelect(vector<int>& a, int l, int r, int index)
{
	int q = randomPartition(a, l, r);
	if (q == index)
	{
		return a[q];
	}
	else
	{
		return q < index ? quickSelect(a, q + 1, r, index) : quickSelect(a, l, q - 1, index);
	}
}
//基于快速排序的选择方法
int findKthLargest(vector<int>& nums, int k)
{
	srand(time(0));
	return quickSelect(nums, 0, nums.size() - 1, nums.size() - k);
}

int main()
{
	vector<int> nums = { 3,2,3,1,2,4,5,5,6 };
	cout << findKthLargest(nums, 4);
	return 0;
}

结果

---

总结