合并果子

合并果子

题意:将n堆果子合并为一堆,每次合并两堆,合并消耗的体力为两堆果子的质量和,求消耗的体力最小。

思路:典型的哈夫曼树,两两最小,再放入优先队列,再重复直到只剩下一堆。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
int n,ans,k;
priority_queue<int,vector<int>,greater<int> > q;
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&k);
        q.push(k);
    }
    for(int i=1;i<n;i++){
        int a=q.top();
        q.pop();
        int b=q.top();
        q.pop();
        ans+=a+b;
        q.push(a+b);
    }
    printf("%d",ans);
    return 0;
}

 

posted @ 2019-07-31 23:08  Ldler  Views(267)  Comments(0Edit  收藏  举报