堆排序

最新推荐文章于 2024-09-04 23:05:17 发布

kai_ITthinker

最新推荐文章于 2024-09-04 23:05:17 发布

阅读量266

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/kai_ITthinker/article/details/64127010

版权

算法专栏收录该内容

11 篇文章 0 订阅

订阅专栏

/* 调整堆， 向下调整的过程 */
	static void heapAdjust(std::vector<int> &vec, int start, int end)
	{
		int i = start;
		while((2*i + 1) < end)
		{			
			 int maxIndex = 2*i + 1;
			 if ((2*i + 2) < end)
			 {
				 if (vec[2*i + 2] > vec[maxIndex])
				 {
					 maxIndex = 2*i + 2;
				 }
			 }
			 if (vec[maxIndex] > vec[i])
			 {
				 int tmp = vec[i];
				 vec[i] = vec[maxIndex];
				 vec[maxIndex] = tmp;
				 
			 }
			 else
			 {
				 break;
			 }
			 i = maxIndex;
		}
	}
/* 建堆，从 vec.size() / 2 - 1开始，选择子树及自身最小的节点为当前的根节点的值，依次处理，直至根节点，全局向上调整，局部向下调整 */
	static void buildHeap(std::vector<int>& vec)
	{
		for (int i = vec.size() / 2 - 1; i >= 0 ; i--)
		{			
			heapAdjust(vec, i, vec.size());
		}
	}
	
	/* 堆排序， 入口，时间复杂度O(nlogn)，大根堆 */
	static void heapSort(std::vector<int>& vec)
	{
		if (vec.empty())
		{
			return;
		}
		buildHeap(vec);
		for (int i = vec.size() - 1; i > 0; i--)
		{
			int tmp = vec[i];
			vec[i] = vec[0];
			vec[0] = tmp;
			heapAdjust(vec, 0, i);
		}
	}