CSU-1588 合并果子

1588: 合并果子

Submit Page      Summary      Time Limit: 1 Sec       Memory Limit: 128 Mb       Submitted: 1351       Solved: 580    

---

Description

现在有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

Hint

Source

国防科学技术大学第十八届银河之光文化节ACM程序设计竞赛初赛

思路：和POJ-3253几乎一样，每次寻找最少的两堆进行合并即可。

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
priority_queue <int,vector<int>,greater<int> > q;
int main()
{
    int t;
    cin >> t;
    while(t--)
    {
        int n;
        int ans=0;
        cin >> n;
        int x;
        while(!q.empty())   q.pop();
        for(int i=0;i<n;i++)
        {
            cin >> x;
            q.push(x);
        }
        while(!q.empty())
        {
            int a=q.top();
            q.pop();
            int b=q.top();
            q.pop();
            ans=ans+a+b;
            q.push(a+b);
            if(q.size()==1)
                break;
        }
        cout << ans << endl;
    }
    return 0;
}