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