通过STL建立二叉堆
STL函数库中的优先队列底层就是二叉堆的建立priority_queue<>;
简单的整数的比较直接用默认的比较方法就行了,如果是结构体或者别的类型需要自己手写比较方法。
int类型的数据默认比较方法是大根堆less
int main()
{
int a[]={5,6,5,1,2,3,8,5,9};
priority_queue<int,vector<int>,greater<int> > pque(a,a+9);//greater从小到大,less为默认从大到小
for(int i=0;i<9;i++)
{
cout<<pque.top();
pque.pop();
}
}
自定义比较类型比较结构体中的数据,手写friend bool operator比较方法
struct node
{
int data;
int Lchild;
int Rchild;
friend bool operator < (node a,node b)//比较函数的重载
{
return a.data>b.data;//data从小到大排序
}
};
int main()
{
priority_queue<node> que;
for(int i=0;i<5;i++)
{
node n;
cin>>n.data;
que.push(n);
}
for(int i=0;i<5;i++)
{
cout<<que.top().data;
que.pop();
}
}
另外贴一下copy别人的代码,手动二叉堆的实现
void HeapAdjust(int a[],int s,int m)//一次筛选的过程
{
int rc,j;
rc=a[s];
for(j=2*s;j<=m;j=j*2)//通过循环沿较大的孩子结点向下筛选
{
if(j<m&&a[j]<a[j+1]) j++;//j为较大的记录的下标
if(rc>a[j]) break;
a[s]=a[j];s=j;
}
a[s]=rc;//插入
}
void HeapSort(int a[],int n)
{
int temp,i,j;
for(i=n/2;i>0;i--)//通过循环初始化顶堆
{
HeapAdjust(a,i,n);
}
for(i=n;i>0;i--)
{
temp=a[1];
a[1]=a[i];
a[i]=temp;//将堆顶记录与未排序的最后一个记录交换
HeapAdjust(a,1,i-1);//重新调整为顶堆
}
}
int main()
{
int n,i;
scanf("%d",&n);
int a[n+1];
for(i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
HeapSort(a,n);
for(i=1;i<=n;i++)
{
printf("%d",a[i]);
}
}