【C++】CLRS上的堆排序实现

14 篇文章 0 订阅
13 篇文章 0 订阅

N天前照着书上伪代码写的堆排序,今天改成了非递归的,自己码出来还是挺涨姿势的……

#include <iostream>
#include <vector>
#include <cstdlib>

typedef std::vector<int> VecInt;

//功能:交换两个整型变量
void Swap(int& a, int& b)
{
	int temp = a;
	a = b;
	b = temp;
}

//功能:建大根堆,非递归
void Max_Heapify(VecInt& arr, int i, int heapsize)
{
	while (true)
	{
		//左右子节点和最大节点下标
		int left = 2 * i;
		int right = 2 * i + 1;
		int largest = 0;

		//最大节点下标设为largest
		if (left <= heapsize && *(arr.begin() + left) > *(arr.begin() + i))
			largest = left;
		else
			largest = i;
		if (right <= heapsize && *(arr.begin() + right) > *(arr.begin() + largest))
			largest = right;

		//最大节点移到子树根上,在该节点原处进行递归
		if (largest != i)
		{
			Swap(*(arr.begin() + largest), *(arr.begin() + i));
			i = largest;
		}
		else
			break;
	}
}

//功能:堆排序,下标0的元素空出
void Heap_Sort(VecInt& arr, int length)
{
	//初始化堆长
	int heapsize = length;

	//从最后一个非叶节点开始向根节点,依次对每个子树整堆
	for (int i = length / 2; i != 0; --i)
		Max_Heapify(arr, i, heapsize);

	//将根节点和最后一个节点变量交换,堆长减一,由根节点开始整堆
	for (int i = length; i != 1; --i)
	{
		Swap(*(arr.begin() + 1), *(arr.begin() + i));
		--heapsize;
		Max_Heapify(arr, 1, heapsize);
	}
}


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值