哈夫曼树,第一行输入一个数n,表示叶结点的个数。需要用这些叶结点生成哈夫曼树,根据哈夫曼树的概念,这些结点有权值,即weight,题目需要输出所有结点的值与权值的乘积之和的最小值。
输入描述:
输入有多组数据。 每组第一行输入一个数n,接着输入n个叶节点(叶节点权值不超过100,2=n=1000)。
输出描述:
输出权值。
示例1
输入
5 1 2 2 5 9
输出
37
#include <iostream>
#include <queue>using namespace std;
//叶子结点到根 最小带权路径之和
int main() {//优先队列存储叶子结点 找最小值priority_queue<int> pqueue;//默认大根堆 取相反数int n;scanf("%d",&n);for(int i=0;i<n;i++){int leaf;scanf("%d",&leaf);//取得pqueue.push(-leaf);//插入}//带权路径和int res=0;//存储带权路径和的中间结果while(pqueue.size()>1){int leaf1=pqueue.top();//取出来pqueue.pop();int leaf2=pqueue.top();pqueue.pop();//计算res=res+leaf1+leaf2;//构成的新子树插入到原集合中pqueue.push(leaf1+leaf2);}printf("%d\n",-res);//构建完成哈夫曼树,输出结果return 0;
}