数据结构实验-堆排序（维护大小根堆）

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


void maxHeapAdjust(int *a, int start, int end)
{
	int cur = start;			//当前结点的位置 
	int left = 2 * cur + 1;		//左孩子的位置 
	int curvalue = a[cur];			//当前结点的具体数值
	for(; left <= end; cur = left, left = 2 * left + 1) {
		//left代表的是左孩子，left+1代表的是右孩子
		if(left < end && a[left] < a[left + 1]) {		//选择左右孩子中较大的进行交换 
			left ++;
		}
		if(curvalue >= a[left]) break;		//如果curvalue 大于左右孩子（max）了，就退出 
		//如果树还是需要被调整 那么就交换父节点和左右子树中的最大值
		a[cur] = a[left];
		a[left] = curvalue; 
	} 
}


void heapSortAsc(int *a, int n)
{
	//调整堆，维护成最大堆 
	for(int i = n / 2; i >= 0; i--) {
		maxHeapAdjust(a, i, n -1);
	}
	 
	//将最大堆的根节点交换到末尾处，实现排序
	for(int i = n - 1; i > 0; i--) {
		swap(a[0], a[i]);
		
		//交换过后继续调整堆
		maxHeapAdjust(a, 0, i - 1); 
	} 
}

//和调整成为最大堆相反 
void minHeapAdjust(int *a, int start, int end) {
	int cur = start;
	int left = 2 * cur + 1;
	int curvalue = a[cur];
	for(; left <= end; cur = left, left = 2 * left + 1) {
		if(left < end && a[left] > a[left + 1]){
			//left表示的是子节点中的比较小的数 
			left ++;
		}
		
			//如果父节点小于子节点的最小值（已经是最小堆结构） 
		if(curvalue <= a[left]) break;
		
			//没有构成最小堆就继续维护
			//交换父节点和子节点中的最小值 
		a[cur] = a[left];			 
		a[left] = curvalue;			 
	}
} 

void heapSortDes(int *a, int n) {
	for(int i = n / 2; i >= 0; i--) {
		minHeapAdjust(a, i, n - 1);
	}
	
	for(int i = n - 1; i > 0; i--) {
		swap(a[0], a[i]);
		
		//交换完毕之后继续维护成为最小堆 
		minHeapAdjust(a, 0 , i - 1);
	}
}



int main()
{
	int a[] = {30, 50, 43, 15, 67, 78, 98, 66, 44};
	int len = sizeof(a) / sizeof(int);
	cout<<"Before: ";
	for(int i = 0; i < len; i++) {
		cout<< a[i] <<" ";
	}
	heapSortAsc(a, len);
	//heapSortDes(a, len);
	cout<<"\nAfter:";
	for(int i = 0; i < len; i++) {
		cout<< a[i] <<" ";
	}
	
}