0.题目说明
在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
这道题可以看成是排序题,我这里借着这道题实现一下快排和堆排序。
1.快速排序
快速排序相关的内容我看的这篇博客,讲得很清楚:
https://blog.csdn.net/adusts/article/details/80882649
https://blog.csdn.net/Krito_D_k/article/details/82500738
思想:
-
在待排序的元素任取一个元素作为基准(通常选第一个元素,但最的选择方法是从待排序元素中随机选取一个作为基准),称为基准元素;
-
.将待排序的元素进行分区,比基准元素大的元素放在它的右边,比其小的放在它的左边;
-
对左右两个分区重复以上步骤直到所有元素都是有序的。
我用第一个博客的思路写的程序如下:
class Solution {
public:
void Quicksort(vector<int>& nums,int left,int right)
{
int i = left, j = right;
while (i!=j)
{
while (nums[j] >= nums[left] && i != j) //最开始移动右边哨兵,若右边哨兵大于等于基准且i和j不相等时,继续移动
{
j--;
}
while (nums[i] <=nums[left] && i != j) //若左边哨兵小于等于基准且i和j不相等时,左边哨兵继续移动
{
i++;
}
//当左边哨兵大于基准时,继续判断
if (i != j) //当两个哨兵没重合时,交换两个哨兵位置,然后继续运动。
{
swap(nums[i], nums[j]);
}
else //当两个哨兵位置重合时,交换当前位置和基准位置,第一次变换结束
{
swap(nums[left], nums[i]);
}
}
if (left<i - 1)
Quicksort(nums, left, i - 1);
if (j + 1<right)
Quicksort(nums, j + 1, right);
}
int findKthLargest(vector<int>& nums, int k) {
int i = 0, j = nums.size() - 1; //i是左边哨兵,j是右边哨兵
Quicksort(nums,i,j);
for (int i = 0; i < nums.size(); i++ )
cout<<nums[i] ;
return nums[nums.size()-k];
}
};
2.堆排序
堆排序可以参考博客:
https://www.cnblogs.com/0zcl/p/6737944.html
https://www.cnblogs.com/wanglei5205/p/8733524.html
堆是一种特殊的树形数据结构,即完全二叉树。堆分为大根堆和小根堆,大根堆为根节点的值大于两个子节点的值;小根堆为根节点的值小于两个子节点的值,同时根节点的两个子树也分别是一个堆。
堆排序基本思路:
- 建立大根堆–将n个元素组成的无序序列构建一个大根堆,
- 交换堆元素–交换堆尾元素和堆首元素,使堆尾元素为最大元素;
- 重建大根堆–将前n-1个元素组成的无序序列调整为大根堆
- 重复执行步骤二和步骤三,直到整个序列有序。
根据这篇博客写的代码如下:
void MaxHeap(vector<int>& arr, int len, int index) //把数组变为最大堆形式
{
int left = index * 2 + 1;
int right = index * 2 + 2;
int maxIdx = index;
if (left<len && arr[left] > arr[maxIdx]) maxIdx = left; //两个判断顺序不能换,否则可能报错
if (right<len && arr[right] > arr[maxIdx]) maxIdx = right;
if (maxIdx != index) //如果交换过位置,则继续
{
swap(arr[maxIdx], arr[index]);
cout << "haha" << endl;
MaxHeap(arr, len, maxIdx);
}
}
void heapSort(vector<int> &arr, int size)
{
for (int i = size / 2 - 1; i >= 0; i--)
{
cout << "i:"<<i << endl;
MaxHeap(arr, size, i); // 构建大根堆(从最后一个非叶子节点向上) (从右到坐,从下到上)
for (int i = 0; i < arr.size(); i++)
cout << arr[i];
}
for (int i = 0; i < arr.size() - 1; i++) //循环arr.size()-1次
{
swap(arr[0], arr[size-1]); //交换当前堆首末位置
//MaxHeap(arr, size, 0); //初始化当前堆
size--;
MaxHeap(arr, size, 0); //初始化当前堆
}
}
这里说明一下MaxHeap函数调用一次之后得到的不是大堆根,调用一次这个函数下面的子树没办法保证达到此条件,所以要从下到上,先让下面的子树满足条件,然后最后再对整个树调用这个函数,比如我们可以随便举个例子:4 6 8 5 9 10 12,读者可以自己试试,要是直接对整个树调用这个函数出来的结果是不是大堆根。
另外说明一下这道题我用堆排比快排耗时少。