nyoj55 懒省事的小明

原题链接http://acm.nyist.net/JudgeOnline/problem.php?pid=55

这道题之前做过，今天重做了一次，各种Wrong Answer，心情十分沮丧。标准程序也过不去，后来看了讨论区才知道测试数据的值偏大（最大结果可以到24亿），所以要用long long类型。

原理:跟huffman树有点相似，使权重(即果实个数)大的离根节点更近，通俗说就是在合并过程中，个数多的堆放到后面再合并。

方法一 用stl中priority_queue优先队列，但是默认为从大到小排序，通过改写比较结构体，模拟最小堆，使队列的首元素权重最小。

方法二 用stl中的multiset，集合中默认排序从小到大，更方便，果然还是不够熟悉。

代码如下：

//方法一 :用priority_queue

#include<stdio.h>
#include<queue>
using namespace std;
struct cmp //因为priority_queue默认是从大到小排序，所以自己定义cmp结构体，使数据从小到大排序
{
    bool operator ()(long long &a,long long &b)
    {
        return a>b;
    }
};
int main()
{
    long long result,i,t,n,tmp;  //为了方便，我这里全部定义为 long long
    scanf("%lld",&t);   
    while(t--)
    {
        priority_queue<long long,vector<long long>,cmp> q; //定义优先级队列
        scanf("%lld",&n);
        for(i=0;i<n;i++)
        {
            scanf("%lld",&tmp);
            q.push(tmp);
        }
        result=0;
        while(q.size()>1) //只剩一堆，跳出循环
        {
            tmp=q.top();
            q.pop();
            tmp=tmp+q.top();
            q.pop();
            q.push(tmp);
            result=result+tmp;
        }
        printf("%lld\n",result);
    }
    return 0;
 }

//方法二 :multiset(改编自nyoj标准程序)

#include<stdio.h>
#include<set>
using namespace std;
int main()
{
    long long n;
    scanf("%lld",&n);
    while(n--)
    {
        long long sum=0;
        multiset<long long> v;
        long long m;
        scanf("%lld",&m);
        while(m--)
        {
            long long temp;
            scanf("%lld",&temp);
            v.insert(temp);
        }
        while(1)
        {
            if(v.size()==1){break;}
            multiset<long long>::iterator it=v.begin();
            long long a=*it;
            long long b=*(++it);
            v.erase(v.begin());
            v.erase(v.begin());
            v.insert(a+b);
            sum=sum+a+b;
        }
        printf("%lld\n",sum);
    }
    return 0;
 }

//小贴士:一开始我以为输入格式采用 %I64d或者%lld应该都可以ac，可是用%I64d输入会超时，应该是oj要求吧。

//不过百度说，一般用到long long类型题目都会给出输入和输出格式。