百练4080:Huffman编码树

BaiLian4080 Huffman编码树

解题思路

所求即为最短路径之和, 也等于新生成节点频率之和.所以可以使用优先队列每次获取一个最小值和次小值, 所求之和即为新生成节点的频率…再依次获取即可.
这个题还用到了priority_queue.使用如下:
Priority_queue:优先队列, 不仅可以存储数据还可以按照某种规则进行排序.每次的push, 和pop操作, 里面数据都会进行动态调整, 把优先级高的放在前面.
使用 :
template <class T, class Container = vector,
class Compare = less > class priority_queue;
T:数据类型, 第三个参数默认不写是从大到小, 如果从小到大可以使用greater.

代码

#include<iostream>
#include<queue>
using namespace std;
priority_queue<int, vector<int>, greater<int> >Q;
int total = 0;
int main()
{
   
	int n,w;
	cin >> n;
	while (n--)
	{
   
		cin >> w;
		Q.push(w);
	}
	while (Q.size() != 1)
	{
   
		int sum = 0;
		sum += Q.top();
		Q.pop();
		sum += Q.top();
		Q.pop();
		Q.push(sum);
		total += sum;
	}
	cout << total << endl;
	return 0;
}