哈夫曼树九度教程第30题优先队列小顶堆

最新推荐文章于 2020-06-06 18:51:09 发布

原创最新推荐文章于 2020-06-06 18:51:09 发布 · 287 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#哈夫曼树

机试指南之路同时被 2 个专栏收录

47 篇文章

订阅专栏

数据结构

2 篇文章

订阅专栏

本文介绍了一种使用优先队列简化哈夫曼树构建并计算所有结点的值与权值乘积之和的方法。通过输入叶结点数量及权值，实现了哈夫曼树的快速构建与权值总和的高效计算。

题目链接

哈夫曼树，第一行输入一个数n，表示叶结点的个数。需要用这些叶结点生成哈夫曼树，根据哈夫曼树的概念，这些结点有权值，即weight，题目需要输出所有结点的值与权值的乘积之和。
输入描述:
输入有多组数据。
每组第一行输入一个数n，接着输入n个叶节点（叶节点权值不超过100，2<=n<=1000）。
输出描述:
输出权值。
示例1
输入
5
1 2 2 5 9
输出
37

解题思路：
典型的求解哈夫曼树，使用优先队列可大大简化代码量。

AC代码：

#include <iostream>
#include <queue> 
#include <functional>
using namespace std;
int main() {
	int n;
	priority_queue<int, vector<int>, greater<int> > Q;
	while (cin >> n) {
		while (Q.empty() == false) Q.pop();
		for (int i = 0; i < n; i++) {
			int x;
			cin >> x;
			Q.push(x);
		}
		int ans = 0;
		while (Q.size() > 1) {
			int a = Q.top();
			Q.pop();
			int b = Q.top();
			Q.pop();
			ans = ans + a + b;
			Q.push(a + b);
		}
		cout << ans << endl;
	}
	return 0;
}