Heap Sort ——重拾堆排序

之前有整理八大排序算法,也有源码和对比,好久不写堆排序了,有点遗忘,重写一遍。

堆排序包括两个过程,建堆和调整堆。

建堆:找到第一个非叶结点(n/2-1)(堆是完全二叉树),然后从底向上进行调整。

// heap sort.cpp : 定义控制台应用程序的入口点。
//

#include "stdafx.h"
#include <algorithm>
using namespace std;

void AdjustMaxHeap(int* array,int pos,int end){//array中除了pos之外均满足堆的定义
//本函数进行调整,使array[pos~n]成为一个大顶堆
	int temp = array[pos];
	for(int i = 2*pos+1;i<=end;i=i*2+1){
		if(i<end && array[i]<array[i+1])
			++i;//i记录较大的下标
		if(temp>array[i])
			break;
		array[pos]=array[i];
		pos = i;		
	}
	array[pos] = temp;//插入最开始不和谐的元素
}

void HeapSort(int* array,int n){
	for(int i = n/2-1;i>=0; i--){ //从第一个非叶结点进行调整
		AdjustMaxHeap(array,i,n-1);
	}
	for(int i = n-1;i>=0;i--){
		swap(array[0],array[i]);
		AdjustMaxHeap(array,0,i-1);
	}
}

int _tmain(int argc, _TCHAR* argv[])
{
	int a[]={5,3,2,1,4,9,2,6};
	HeapSort(a,8);
	return 0;
}


你可能感兴趣的:(堆排序,排序算法)