C++堆排序简单实现

前言

最近在复习的时候,复习到了堆排序,简单了解后,做一个代码总结。

注:本文章只对功能做一个实现。参考文献贴在文章末尾。

完整代码

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int a[N];
int n;

void print(){
    
    
	for(int i = 0; i < n; i++) cout << a[i] << ' ';
	cout << '\n';
}

//调整大顶堆 
void adjust(int fa, int len){
    
    
	//记录左子节点,右子节点 
	int l = fa * 2 + 1, r = fa * 2 + 2; 
	//最大节点 
	int largest = fa;
	//如果左子节点大于最大节点,就修改最大节点 
	if(l < len && a[l] > a[largest]) {
    
    
		largest = l;
	}
	//如果右子节点大于最大节点,就修改最大节点 
	if(r < len && a[r] > a[largest]) {
    
    
		largest = r;
	}
	//如果最大节点不是父节点,那么就交换父节点和最大节点 
	if(largest != fa){
    
    
		swap(a[fa], a[largest]);
		//继续调整最大节点 
		adjust(largest, len);
	}
}

//建立大顶堆 
void buildHeap(int len){
    
    
	//从最后一个非叶子节点开始,进行调整 
	for(int i = len / 2 - 1; i >= 0; i--){
    
    
		adjust(i, len);
	}
}

void HeapSort(){
    
    
	//首先建立大顶堆 
	buildHeap(n);
	int len = n;
	for(int i = len - 1; i >= 0; i--){
    
    
		//此时已经是大顶堆,只需要把堆顶元素和最后一个元素交换就可以了 
		swap(a[0], a[i]);
		//调整堆顶元素,保持大顶堆 
		//因为交换之前是大顶堆
		//也就是所有根节点比左右子节点大
		//那么只需要调整交换之后的根节点就可以了 
		adjust(0, i);
	}
}

int main(){
    
    
	cin >> n;
	for(int i = 0; i < n; i++) cin >> a[i];
	HeapSort();
	for(int i = 0; i < n; i++) cout << a[i] << ' ';
	return 0;
} 

参考文献

  1. 菜鸟教程:堆排序
  2. 谈谈堆排序,大顶堆,小顶堆?
  3. 图解排序算法(三)之堆排序

猜你喜欢

转载自blog.csdn.net/qq_52855744/article/details/123620671