# 题目描述
哈夫曼树,第一行输入一个数 n,表示叶结点的个数。需要用这些叶结点生成哈夫曼树,根据哈夫曼树的概念,这些结点有权值,即 weight,题目需要输出所有叶子结点的路径长度与权值的乘积之和。
# 输入
输入有多组数据。
每组第一行输入一个数 n,接着输入 n 个叶节点(叶节点权值不超过 100,2<=n<=1000)。
# 输出
输出权值。
# 样例输入
2 | |
2 8 | |
3 | |
5 11 30 |
# 样例输出
10 | |
62 |
# 题解
#include<bits/stdc++.h> | |
using namespace std; | |
priority_queue<int, vector<int>, greater<int> > q; | |
int main() { | |
int n; | |
while (cin >> n) { | |
while (!q.empty()) { | |
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 += a + b; | |
q.push(a + b); | |
} | |
cout << ans << endl; | |
} | |
return 0; | |
} |