堆排序笔记

还是记录到博客上不易找不到笔记

堆排序原理

堆排序是利用堆这种数据结构对无序序列进行排序，属于贪婪算法，先利用大根（或小根堆后面省略）堆数据结构找出序列中的最大值，然后将最大值从序列中移出，然后再利用堆数据结构在剩余序列中将第二大值找出，以此类推直至所有数据都排序。

堆排序的难点

• 给定的序列怎么建立成大根堆

堆排序需要牢记的几个重要知识点

• 第一个非叶子节点位置：len/2-1;
• 左子树位置：2*i+1;
• 右子树位置：2*i+2;

代码

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

// 递归方式构建大根堆(len是arr的长度，index是第一个非叶子节点的下标)
void adjust(vector<int> &arr, int len, int index)
{
	int left = 2 * index + 1; // index的左子节点
	int right = 2 * index + 2;// index的右子节点

	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]);
		adjust(arr, len, maxIdx);
	}

}

// 堆排序
void heapSort(vector<int> &arr, int size)
{
	// 构建大根堆（从最后一个非叶子节点向上）
	for (int i = size / 2 - 1; i >= 0; i--)
	{
		adjust(arr, size, i);
	}

	// 调整大根堆
	for (int i = size - 1; i >= 1; i--)
	{
		swap(arr[0], arr[i]);           // 将当前最大的放置到数组末尾
		adjust(arr, i, 0);              // 将未完成排序的部分继续进行堆排序
	}
}

int main()
{
	vector<int> arr = { 9,8,7,6,5,4,3,2,1 };
	cout << "排序前" << endl;
	for (int i = 0; i < arr.size(); i++)
	{
		cout << arr[i] << " ";
	}
	cout << endl;


	heapSort(arr, arr.size());
	cout << "排序后" << endl;
	for (int i = 0; i < arr.size(); i++)
	{
		cout << arr[i]<<" ";
	}
	cout << endl;
	return 0;
}

输出

排序前
9 8 7 6 5 4 3 2 1
排序后
1 2 3 4 5 6 7 8 9