AcWing 148 合并果子 题解 (贪心—Huffman树)
作者:互联网
思路:按照树的思想考虑合并,树的最底层进行合并时需要计算的次数最多,所以质量最小的两个果子应该在树的最底层,最终所有果子合并到树的根节点,根节点的值即为消耗的体力和。将所有果子质量存入堆中(实现排序),每次挑出质量最小的两堆果子(堆顶的两个元素)合并,将总体力数加上这两堆果子的质量,将新合并成的果子堆的质量存入堆中(作为新的树的节点),这样就能实现合并消耗的总体力最小
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
int main()
{
int n;
priority_queue<int , vector<int>, greater<int>>heap;//用优先队列实现对所有果子的质量排序的操作
cin>>n;
while(n -- ){
int a;
cin>>a;
heap.push(a);
}
int res = 0;//记录答案
while(heap.size() > 1){//当树的节点数大于1,代表果子还没合并完
//取出当前最小的两个果子
int a = heap.top(); heap.pop();
int b = heap.top(); heap.pop();
res += a + b;//合并消耗的体力加上a和b的质量
heap.push(a + b);//将这两个果子合并,合并后的质量和作为新的节点加入树上
}
cout<<res<<endl;
return 0;
}
标签:果子,int,题解,合并,148,质量,heap,节点,AcWing 来源: https://blog.csdn.net/qiaodxs/article/details/120494851