jzoj1082-合并果子【堆,贪心】

题目

有n堆大小不同的果子堆，每次合并两堆消耗它们两堆的重量之和，求把它们全部合并的最小消耗

---

输入

输入包括两行，第一行是一个整数n(1<＝n<=10000)，表示果子的种类数。第二行包含n个整数，用空格分隔，第i个整数ai(1<＝ai<=20000)是第i种果子的数目。

输出

输出包括一行，这一行只包含一个整数，也就是最小的体力耗费值。输入数据保证这个值小于2^31。

样例输入

3
1 2 9

样例输出

15

---

解题思路

贪心就是每次都合并最小的两个，这里用堆可以方便一些

---

代码

#include
using namespace std;
int a[10001],num,x,n;
long long s,u;
void up(int x)
{
    int t;
    while (x>1 && a[x]2])
    {
        t=a[x];a[x]=a[x/2];a[x/2]=t;
        x/=2;
    }
}//调整堆
void down(int x)
{
    int t,y;
    while (x*2<=num && a[x]>a[x*2] || x*2+1<=num && a[x]>a[x*2+1])
    {
        y=x*2;
        if (x*2+1<=num && a[x*2]>a[x*2+1]) y++;
        t=a[x];a[x]=a[y];a[y]=t;
        x=y;
    }
}//调整堆
void insert(int x)
{a[++num]=x;up(num);}//插入元素
int main()
{
    scanf("%d",&n); 
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        insert(x);
    }//建堆
    while (num>1)
    {
        u=a[1];a[1]=a[num];num--;down(1);
        u+=a[1];a[1]=a[num];num--;down(1);//取两个最小的
        s+=u;num++;a[num]=u;up(num);//合并后放回
    }
    printf("%d",s);
}