7-2 哈夫曼树

哈夫曼树，第一行输入一个数n，表示叶结点的个数。

需要用这些叶结点生成哈夫曼树，根据哈夫曼树的概念，这些结点有权值，即weight，题目需要输出哈夫曼树的带权路径长度（WPL）。

输入格式:

第一行输入一个数n，第二行输入n个叶结点（叶结点权值不超过1000，2<=n<=1000）。

输出格式:

在一行中输出WPL值。

输入样例:

5
1 2 2 5 9

输出样例:

37

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

 

如图为构造出的哈夫曼树。这里给出WPL计算方法：

WPL = 所有叶子结点的权值乘以路径长度之和=所有非叶子结点权值之和

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

long long  buildHuffmanTree(int leafNodes[], int n) {
	long long wpl = 0;
	while (n > 1) {
		int min1 = 0, min2 = 1;
		if (leafNodes[min1] > leafNodes[min2]) {
			swap(min1, min2);
		}
		for (int i = 2; i < n; i++) {
			if (leafNodes[i] < leafNodes[min1]) {
				min2 = min1;
				min1 = i;
			} else if (leafNodes[i] < leafNodes[min2]) {
				min2 = i;
			}
		}//寻找第一小权值和第二小权值的叶子结点
		int newNode = leafNodes[min1] + leafNodes[min2];
		wpl += newNode;
		leafNodes[min1] = newNode;
		leafNodes[min2] = leafNodes[n - 1];
		n--;//将 newNode 添加到权值数组中，并将 min2 节点的权值替换为数组中的最后一个元素。最后，将节点数 n 减1，表示两个叶子节点合并为一个新节点//
	}
	return wpl;
}

int main() {
	int n;
	cin >> n;
	int leafNodes[n];
	for (int i = 0; i < n; i++) {
		cin >> leafNodes[i];
	}
	int result = buildHuffmanTree(leafNodes, n);
	cout << result << endl;
	return 0;
}