九度 1172：哈夫曼树

题目描述：

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

 

思路

1. 构建哈夫曼树然后 topDown 遍历即可

2. 也可以不显式的创建树. 所有树叶点的和即为所求(根节点以外)

 

代码 STL priority_queue 自定类型不会用

#include <iostream> #include <stdio.h> #include <queue>
using namespace std; class Node { public: int value; Node *left, *right; Node(int _value) { value = _value; } Node() { Node(0); } bool operator<(const Node &ths) const { return this->value < ths.value; } }; int n; int res; void calculate(Node *root, int depth) { if(root->left == NULL && root->right == NULL) { res += root->value*depth; return; } calculate(root->left, depth+1); calculate(root->right, depth+1); } int main() { freopen("testcase.txt", "r", stdin); while(scanf("%d", &n) != EOF) { res = 0; priority_queue<Node*> record; int val; for(int i = 0; i < n; i ++) { scanf("%d", &val); record.push(new Node(val)); } while(record.size() != 1) { Node *node1 = record.top(); record.pop(); Node *node2 = record.top(); record.pop(); Node *newNode = new Node(node1->value + node2->value); newNode->left = node1; newNode->right = node2; record.push(newNode); } calculate(record.top(), 0); cout << res << endl; } return 0; }