poj3253 Fence Repair 哈夫曼树

最新推荐文章于 2021-07-26 11:36:28 发布

原创最新推荐文章于 2021-07-26 11:36:28 发布 · 240 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一个基于优先队列的数据结构应用问题，并通过C++实现了解决方案。主要思路是将输入的整数使用优先队列进行存储，之后不断取出最小的两个数进行合并并重新插入队列，直到队列只剩下一个元素，该过程的总代价即为答案。

题目连接

#include<iostream>
#include<string>
#include<queue> 
using namespace std;
int main(){
	int n, m, x, y;
	while(cin>>n){
		priority_queue<int,vector<int>,greater<int> > q;
	
		for(int i=0;i<n;i++){
			cin>>m;
			q.push(m);
		}
		long long sum = 0;
		if(q.size()==1){//只有一段 
			sum+=q.top();
			q.pop();
		}
		while(q.size()>1){
			x = q.top();
			q.pop();
			y = q.top();
			q.pop();
			sum+=(x+y);
			q.push(x+y);
		}
		cout<<sum<<endl;
	}
}