其他分享
首页 > 其他分享> > AcWing 148 合并果子 题解 (贪心—Huffman树)

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