C -合并果子
Crawling in process...
Crawling failed
Time Limit:1000MS Memory Limit:131072KB64bit IO Format:%lld & %llu
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
优先队列的应用,实际上就是相当于一种哈夫曼算法,每次取最小的两堆进行合并,将合并后的那堆同样放入其中进行大小比较。关于优先队列,可以参考http://blog.csdn.net/qq_30076791/article/details/50569241
#include<stdio.h>
#include<queue>
#include<functional>
using namespace std;
int main()
{
int t,n,a;
scanf("%d",&t);
while(t--)
{
priority_queue<int,vector<int>,greater<int> >q;
scanf("%d",&n);
for(int i=0; i<n; i++)
{
scanf("%d",&a);
q.push(a);
}
int ans=0,tmp;
while(q.size()>=2)
{
tmp=0;
tmp+=q.top();
q.pop();
tmp+=q.top();
q.pop();
ans+=tmp;
q.push(tmp);
}
printf("%d\n",ans);
}
return 0;
}
第一次使用优先队列做题,其中有两点需要注意:
【1】priority_queue<int,vector<int>,greater<int> >q;中的最后两个>>中间要有一个空格;
<span style="font-family: Arial, Helvetica, sans-serif; font-size: 12px;">【2】取队列中的元素是q.top(),而不是q.front()。</span>