堆排序(heapsort)

#include<iostream>
#include<stdio.h> 
using namespace std;
/**
创建一个堆:大的数是根 
*/ 
void swap(int arr[],int i,int j)
{
	int temp= arr[i];
	arr[i]=arr[j];
	arr[j]=temp;
}
void heapify(int tree[],int n,int i)
{
	//检查输入的这个节点的位置对不对
	//如果大的数在底层就让它上去 
	if(i>=n)
	{
		return;
	}
	int c1 = 2*i+1;
	int c2 = 2*i+2;
	//计算两个儿子的编号 
	int max = i;
	if(c1<n && tree[c1]>tree[max])
	{
		max=c1;
	}
	if(c2<n && tree[c2]>tree[max])
	{
		max=c2;
	}
	//查找儿子们和父亲中那个大 
	if(max!=i)
	{
		swap(tree,max,i);
		heapify(tree,n,max);
	}
	//如果儿子大于父亲就交换
	//然后继续往下一层判断 
}
void build_heap(int tree[],int n) 
{
	//对每一节点都检查一遍就成功建立成了一个堆 
	int last_node = n-1;
	int parent=(last_node-1)/2;
	for(int i=parent;i>=0;--i)
	{
		heapify(tree,n,i);
	}
}
/*
堆排序:从大到小 
*/
void heap_sort(int tree[],int n)
{
	build_heap(tree,n);
	for(int i=n-1;i>=0;--i)
	{
		swap(tree,i,0);//交换根节点和最后一个叶子 
		heapify(tree,i,0);//砍断那个最后的叶子 
	}
}
int main()
{
	int tree[]={4,10,2,5,1,3};
	int n=6;
	//堆排序 
	heap_sort(tree,n);
	
	for(int i=0;i<n;++i)
	{
		cout<<tree[i]<<" "<<endl;
	 } 
	return 0;
} 

 


来自灯神的讲解,这里我加了一点自己的注释:https://www.bilibili.com/video/av47196993

 

posted @ 2019-10-31 13:18  东坡肉肉君  阅读(179)  评论(0编辑  收藏  举报