2016复旦大学计算机复试上机第三题——求哈夫曼编码的最短长度

2016复旦大学计算机复试上机第三题——求哈夫曼编码的最短长度

题目：给定一个字符串，求哈夫曼编码的最短长度：
思路1：统计每个字符出现的字数，然后构建哈弗曼树，计算每个叶子的权值和路径长度的乘积之和。
思路2：（简单方法）：你会发现就是求哈夫曼树的非根结点权值之和，或者所有非叶结点之和等于所求长度。

思路二实现方法：
priority_queue 优先队列：

priority_queueq; 默认为数字（字典序）大的值在队首top，等价于priority_queue<int, vector, less >q;
priority_queue<int, vector, greater >q; 表示数字（字典序）小的在队首
没有front、back函数，只有top用于读取队首元素

1. 输入数据 abbcccdddd
2. 输出 19

#include <iostream>
#include <string>
#include <cstring>
#include <queue>
#include <map>
using namespace std;
const int maxn = 100;
string str;
map<char, int> mp;
priority_queue<int, vector<int>, greater<int> >q;

int main(){
    getline(cin, str);
    int len = str.length();
    for(int i = 0; i < len; i++){
       if(mp.find(str[i]) == mp.end()){
            mp[str[i]] = 1;
       }
       else{
            mp[str[i]]++;
       }
    }
    for(map<char, int>::iterator it = mp.begin(); it != mp.end(); it++){
        q.push(it->second);
    }
    int ans = 0;
    while(q.size() != 1){
        int a, b, temp;
        a = q.top();
        q.pop();
        b = q.top();
        q.pop();
        temp = a + b;
        ans += temp;
        q.push(temp);
    }
    cout << ans << endl;
    return 0;
}