合并果子（multiset）

合并果子

Time Limit: 1 Sec  Memory Limit:
128 MB

Submit: 312  Solved: 113

[Submit][Status][Web
Board]

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

代码：

#include<cstdio>
#include<set>
using namespace std; 

multiset<int> mst; 

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        mst.clear();
        int n;
        int ans=0;
        scanf("%d",&n);
        for(int i=1;i<=n;i++)
        {
            int x;
            scanf("%d",&x);
            mst.insert(x);
        }
        int anss;
        multiset<int>::iterator it1;
        multiset<int>::iterator it2;
        while(mst.size()>1)
        {
            it1=it2=mst.begin();
            it2++;
            anss=*it1+*it2;
            ans=ans+anss;
            mst.erase(mst.begin());
            mst.erase(mst.begin());
            mst.insert(anss);
        }
        printf("%d\n",ans);
    }
    return 0;
}