哈夫曼树 计算最小WPL

题目描述

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

输入描述:

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

输出描述:

输出权值。

示例1

输入

复制

5  
1 2 2 5 9

输出

复制

37

 

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

// 只有这样，top出来的才不是max，而是min。不知道为什么
//背过！      vector<int>   greater<int>    vg vg vg vg vg vg vg
priority_queue<int,vector<int>,greater<int> > pq;



int main() {
	int n;
	while (cin >> n ) {
		while (pq.size() != 0) {
			pq.pop();
		}


		int t;
		for (int i = 1; i <= n; i++) {
			cin >> t;
			pq.push(t);
		}

		int weight = 0;
		while(pq.size() != 1) {
			int a = pq.top();
			pq.pop();
			int b = pq.top();
			pq.pop();

			weight += (a+b);  // a  与 b 是权重  路径是1  加了之后  weight就全了  把全部路径都加了
			pq.push(a+b);
		}

		cout << weight << endl;
		pq.pop();

	}





}