洛谷_1090 合并果子

题意：

    有n堆果子，我们每次合并其中的两堆所需的体力为它们数目的和，我们现在要求合并给出的n堆果子最少需要多少体力。

思路：

    最少的体力为前2堆最小数目的果子，我们可以用每次排序来求，但我这里用的是堆，先建堆，每次取出一个根并删除，取两个加到答案中，并把它们的和扔到堆里面，直到只剩一堆就退出循环。

代码：

#pragma GCC optimize(2)
#include<cstdio>
int ans,n,num,a[10001];
void swap(int &a,int &b)//在变量前加&可以改变回值
{
	int t=0;
	t=a;a=b;b=t;
}
void up(int x)
{
	int t=0;
	while (x>1&&a[x]<a[x/2])//如果可以往上放，我们就放
	{
		t=a[x];a[x]=a[x/2];a[x/2]=t;x/=2;
	}
}
int get()
{
	int re=a[1],next,now;
	a[1]=a[num];
    num--;
    now=1;
    while(now*2<=num)//把新的根和它最小的子节点替换
    {
    	next=now*2;
    	if (next<num&&a[next]>a[next+1]) next+=1;
		if (a[next]>a[now]) break;
		swap(a[now],a[next]);
		now=next; 
    }
  	return re;//返回之前的根
}

int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		up(i);//建堆
    }num=n;
    while (num!=1)
    {
    	int x=get(),y=get();//取出两个最小的并删除
    	ans=ans+x+y;
    	a[++num]=x+y; up(num);//把它们的和扔回堆里面
    }
    printf("%d",ans);
}