合并果子

Posted 2022-06-01 2462478392lee

合并果子

题意：将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;