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