堆排 luogu1090合并果子

此题为堆排的简单运用

#include<cstdio>
#include<iostream>
using namespace std;
int ans,s,hp[10005],n,siz,cnt;
void pus(int x)//输（插）入 
{
cnt++;
hp[cnt]=x;
int now=cnt;
int fa=cnt/2;
while(now>1)
{
if(hp[now]>=hp[fa])
return;
else
{
swap(hp[now],hp[fa]);
now=fa,fa/=2;
}
}
}
int del()//删除堆顶 维护堆 
{
int res=hp[1];
swap(hp[1],hp[cnt]);
cnt--;
int now=1;
while(2*now<=cnt)//有儿子 
{
int lison=2*now;
if(lison<cnt && hp[lison+1]<hp[lison])//有右儿子
lison++;
if(hp[now]>hp[lison])
swap(hp[now],hp[lison]),now=lison;
else break;
}
return res;
} 
int main()
{
cin>>n;
int a;
for(int i=1;i<=n;i++)
{
cin>>a;
pus(a);//建堆 
}
for(int i=1;i<n;i++)
{
int a1=del();//提取两个最小值
int a2=del();
int s=a1+a2;
ans+=s;
pus(s); 
}
cout<<ans<<endl;
return 0;
}