【LeetCode】13.快速排序和堆排序 (215 数组中的第K个最大元素)

19 篇文章 0 订阅

0.题目说明

在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
在这里插入图片描述
这道题可以看成是排序题,我这里借着这道题实现一下快排和堆排序。

1.快速排序

快速排序相关的内容我看的这篇博客,讲得很清楚:
https://blog.csdn.net/adusts/article/details/80882649
https://blog.csdn.net/Krito_D_k/article/details/82500738

思想:

  1. 在待排序的元素任取一个元素作为基准(通常选第一个元素,但最的选择方法是从待排序元素中随机选取一个作为基准),称为基准元素;

  2. .将待排序的元素进行分区,比基准元素大的元素放在它的右边,比其小的放在它的左边;

  3. 对左右两个分区重复以上步骤直到所有元素都是有序的。

我用第一个博客的思路写的程序如下:

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

堆是一种特殊的树形数据结构,即完全二叉树。堆分为大根堆和小根堆,大根堆为根节点的值大于两个子节点的值;小根堆为根节点的值小于两个子节点的值,同时根节点的两个子树也分别是一个堆。

堆排序基本思路:

  1. 建立大根堆–将n个元素组成的无序序列构建一个大根堆,
  2. 交换堆元素–交换堆尾元素和堆首元素,使堆尾元素为最大元素;
  3. 重建大根堆–将前n-1个元素组成的无序序列调整为大根堆
  4. 重复执行步骤二和步骤三,直到整个序列有序。

根据这篇博客写的代码如下:

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,读者可以自己试试,要是直接对整个树调用这个函数出来的结果是不是大堆根。
另外说明一下这道题我用堆排比快排耗时少。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值