为了掌握简单的STL内容,遂学习优先队列后,做一题以加强记忆~
题意:合并果子,相信很多人都知道了,将n堆果子合并,每合并任意两堆,耗费为这两堆的果子数目,求最小花费将其合并为一堆。
直接优先队列搞起。
#include<cstdio>
#include<cstdlib>
#include<string>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#define maxn 100006
using namespace std;
struct cmp{
bool operator ()(int &a,int &b){
return a > b;
}
};
int main()
{
int n;
while(~scanf("%d",&n))
{
priority_queue<int,vector<int>,cmp> que;
int temp;
for(int i=0;i<n;i++)
{
scanf("%d",&temp);
que.push(temp);
}
int ans = 0;
while(que.size() > 1)
{
int a1 = que.top(); que.pop();
int a2 = que.top(); que.pop();
ans += a1 + a2;
que.push(a1+a2);
}
printf("%d\n",ans);
}
system("pause");
return 0;
}