哈夫曼树【北京邮电大学复试上机题】【优先队列】

题目描述

哈夫曼树，第一行输入一个数n，表示叶结点的个数。需要用这些叶结点生成哈夫曼树，根据哈夫曼树的概念，这些结点有权值，即weight，题目需要输出所有结点的值与权值的乘积之和的最小值。

输入描述:

输入有多组数据。
每组第一行输入一个数n，接着输入n个叶节点（叶节点权值不超过100，2<=n<=1000）。

输出描述:

输出权值。

示例1

输入

复制

5  
1 2 2 5 9

输出

复制

37

哈夫曼树的带权路径长度为数中所有叶子节点的带权路径之和，在计算的时候，结合图可知，在每次加新节点的时候直接answer+=a+b即可求出

#include<iostream>
#include<queue>
using namespace std;

priority_queue<int,vector<int>,greater<int>> q;   //优先队列默认大顶堆，这里把它转成小顶堆 
int main(){
	int n,x;
	while(scanf("%d",&n)!=EOF){
		while(n--){
			cin>>x;
			q.push(x);
		}
		int answer=0;
		while(q.size()>1){
			int a=q.top();
			q.pop();
			int b=q.top();
			q.pop();
			q.push(a+b);
			answer+=a+b;   //求WPL 
		} 
		cout<<answer<<endl;
	}
	return 0;
}