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类型题目都会给出输入和输出格式。