POJ 1521 Huffman树

POJ 1521 Huffman树

本题给出一个字符串，求其Huffman编码的长度。

我们可以统计每个字符出现的次数作为该字符的权重，那么编码总长度就等于外部带权路径长度。进一步地，有结论：Huffman树的外部带权路径长度等于除根结点外的所有结点的权值之和（暂时还没想清楚为什么），因此不需建树，统计所有结点权重即可。

输入仅有一个字符需要特判。

poj不支持c++11，auto不可用。

代码：

#include <iostream>
#include <map>
#include <vector>
#include <string>
#include <queue>
#include <iomanip>
#include <algorithm>

using namespace std;

int main(){
    while (true) {
        string s;
        map<char,int> w;
        cin>>s;
        if(s == "END") break;
        
        sort(s.begin(), s.end());
        w[s[0]] = 1;
        for(int i = 1;i < s.size();i++){
            if(s[i] == s[i-1]){
                w[s[i]]++;
            }else{
                w[s[i]] = 1;
            }
        }
        
        priority_queue<int,vector<int>,greater<int> > pq;
        for(map<char,int>::iterator ite = w.begin();ite != w.end();ite++){
            pq.push(ite->second);
        }
        
        if(pq.size() == 1) cout<<s.size()*8<<" "<<pq.top()<<" "<<setiosflags(ios::fixed)<<setprecision(1)<<(double)s.size()*8.0/(double)pq.top()<<endl;
        else{
            int sum = 0;
            while(pq.size() >= 2){
                int a = pq.top();pq.pop();
                int b = pq.top();pq.pop();
                sum += a+b;
                pq.push(a+b);
            }
            cout<<s.size()*8<<" "<<sum<<" "<<setiosflags(ios::fixed)<<setprecision(1)<<(double)s.size()*8.0/(double)sum<<endl;
        }
    }
}

0ms 192KB

Huffman树结论的证明

Huffman树带权路径和等于其除根结点外所有结点的权重之和。

证明其实很好理解，针对每一个带权路径，其途经的各个结点的权值一定有一部分是这个叶结点贡献的，把一路上这个叶结点的贡献值加起来，就是这条加权路径的长度。而所有的加权路径的长度之和刚好是除根结点外所有结点的权值之和。