Huffman编码树

题目:openjudge 6939

http://cqsyz.openjudge.cn/hanjai/7/

总时间限制: 1000ms

内存限制: 65536kB

描述

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

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

Wi:每个节点的权值。

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

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

输入第一行输入一个整数n,外部节点的个数。第二行输入n个整数,代表各个外部节点的权值。
2<=N<=100输出输出最小外部路径长度总和。样例输入
4
1 1 3 5
样例输出
17

构造哈夫曼树,再统计带权路径和。

#include 
#include 
#include 
using namespace std;
typedef struct node{
    int l,r;
    int p,w,fa;
    bool operator<(const node &x) const{
        return w>x.w;
    }
}node;
node tr[220];
int main(){
    priority_queue q;
    int n,w,t;
    while(cin>>n){
        memset(tr,0,sizeof(tr));
        t=0;
        while(n--){
           cin>>w;
            ++t;
           tr[t]=node{0,0,t,w,0};
           q.push(tr[t]);
        }
        while(q.size()>1){
            node x1=q.top();
            q.pop();
            node x2=q.top();
            q.pop();
            ++t;
            tr[t]=node{x1.p,x2.p,t,x1.w+x2.w,0};
            tr[x1.p].fa=t;
            tr[x2.p].fa=t;
            q.push(tr[t]);
        }
        int ans=0;
        for(int i=1;i<=t;i++){
            if(tr[i].l==0){
                int dep=0;
                node x=tr[i];
                while(x.fa!=0){
                    dep++;
                    x=tr[x.fa];
                }
                ans+=dep*tr[i].w;
            }
        }
        cout<

你可能感兴趣的:(Huffman编码树)