poj 3253 HLG 2059

题意：输入n，再输入n个数，一次可以合并2个数，合并一次需要的能量是两个数的和，求最小能量和是多少？

解题思路：贪心，每次取最短的两个数求和，在加入数组中，在排序，直至只剩一个数为止；

sort 超时了，只能用堆排序了,

1.数组模拟堆

#include<stdio.h>
#include<string.h>
int a[10001],n;
void down(int m)
{
    int t;
    while (m*2<n)
    {
        m*=2;
        if (m<n&&a[m]>a[m+1])
           m++;
        if (a[m]<a[m/2])
        {
            t=a[m];
            a[m]=a[m/2];
            a[m/2]=t;
        }
        else
           break;
    }
}
void built(int n)
{
    int i;
    for (i=n/2;i>0;--i)
      down(i);
}
int main()
{
    while (scanf("%d",&n)!=EOF)
    {
        int x,y,i;
        long long int ans=0;
        memset(a,0,sizeof(a));
        for (i=1;i<=n;i++)
          scanf("%d",&a[i]);
        built(n);
        while (n>1)
        {
            x=a[1];
            a[1]=a[n];
            down(1);
            n--;
            ans=ans+x+a[1];
            a[1]+=x;
            down(1);
        }
        printf("%lld\n",ans);
    }
    return 0;
}

2.stl堆

#include<algorithm>
#include<stdio.h>
using namespace std;
const int maxn=10005;
int a[maxn];
bool cmp(int x,int y)
{
    return x>y;
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        long long int ans=0;
        for(int i=0; i<n; i++)
            scanf("%d",&a[i]);
        make_heap(a,a+n,cmp);
        while(n>1)
        {
            pop_heap(a,a+n,cmp);
            pop_heap(a,a+n-1,cmp);
            a[n-2]+=a[n-1];
            ans+=a[n-2];
            n--;
            push_heap(a,a+n,cmp);
        }
        printf("%lld\n",ans);
    }
    return 0;
}