每日一题 KY188 哈夫曼树

哈夫曼树，第一行输入一个数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;
}