POJ 5941 Huffman编码树｜最小堆

 

题目描述

总时间限制:  1000ms 内存限制: 65535kB

描述

构造一个具有n个外部节点的扩充二叉树，每个外部节点Ki有一个Wi对应，作为该外部节点的权。使得这个扩充二叉树的叶节点带权外部路径长度总和最小：

                                     Min( W1 * L1 + W2 * L2 + W3 * L3 + … + Wn * Ln)

Wi:每个节点的权值。

Li:根节点到第i个外部叶子节点的距离。

编程计算最小外部路径长度总和。

 

输入

第一行输入一个整数t，代表测试数据的组数。
对于每组测试数据，第一行输入一个整数n，外部节点的个数。第二行输入n个整数，代表各个外部节点的权值。
2<=N<=100

输出

输出最小外部路径长度总和。

样例输入

2
3
1 2 3
4
1 1 3 5

样例输出

9
17

问题解决

使用优先队列，使用方法如下：

//小顶堆
priority_queue <int,vector<int>,greater<int> > q;
//大顶堆
priority_queue <int,vector<int>,less<int> >q;

每次取两个最小的节点，和加上两节点值之和，再将其合并加入队列，直到队列中只剩下一个节点为止：

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

int main(){
    int N,n,tmp;
    cin>>N;
    while(N--){
        cin>>n;
        priority_queue<int,vector<int>,greater<int> > q;
        int res=0,x,y;
        for(int i=0;i<n;i++){
            cin>>tmp;
            q.push(tmp);
        }
        while(q.size()>1){
            x=q.top();q.pop();
            y=q.top();q.pop();
            res+=(x+y);
            q.push(x+y);
        }
        cout<<res<<endl;
    }
    return 0;
}