合并果子

合并果子

现在有n堆果子，第i堆有ai个果子。现在要把这些果子合并成一堆，每次合并的代价是两堆果子的总果子数。求合并所有果子的最小代价。

Input
第一行包含一个整数T（T<=50），表示数据组数。
每组数据第一行包含一个整数n（2<=n<=1000），表示果子的堆数。
第二行包含n个正整数ai（ai<=100），表示每堆果子的果子数。

Output
每组数据仅一行，表示最小合并代价。

Sample Input
2
4
1 2 3 4
5
3 5 2 1 4
Sample Output
19
33

AC代码：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
bool cmp(int a,int b)
{
    //a代表的是左边的数（也可以认为左边是较小的一边），那左边大于右边那就是从小到大排序
    return a>b;
}
int main()
{
    int n,m,i,k,x,y,cnt;
    vector<int> a;
    cin >> n;
    while (n--)
    {
        cnt=0;
        cin >> m;
        for (i=0;i<m;++i)
        {
            cin >> k;
            a.push_back(k);
        }
        make_heap(a.begin(),a.end(),cmp);//建立最小树
        while (a.size())
        {
            x=a.front();y=0;pop_heap(a.begin(),a.end(),cmp);a.pop_back();//y必须每次都为0
            if(a.size()){y=a.front();pop_heap(a.begin(),a.end(),cmp);a.pop_back();}
            cnt+=x+y;
            if(a.size()){a.push_back(x+y);push_heap(a.begin(),a.end(),cmp);}
        }
        cout << cnt << endl;
        a.clear();
    }
    return 0;
}