HDU - 2527 (哈夫曼树 + 优先队列)

问题描述：

哈夫曼编码(Huffman Coding)；一个字母的权值等于该字母在字符串中出现的频率。所以Javac++ 想让你帮忙，给你安全数值和一串字符串，并让你判断这个字符串是否是安全的？

Input：

输入有多组case，首先是一个数字n表示有n组数据，然后每一组数据是有一个数值m(integer)，和一串字符串没有空格只有包含小写字母组成！

Input：

如果字符串的编码值小于等于给定的值则输出yes，否则输出no。

SampleInput：

2
12
helloworld
66
ithinkyoucandoit

Sample Output：

no
yes

哈夫曼树模板题，直接判断字符串的WPL值是否超过给定安全值即可。由于优先队列是默认以升序规则(less)排列，每次取数是取出的最大的那个数，所以我们在定义有限队列的时候，将优先队列中的排序规则换成greater.

#include <vector>
#include <string>
#include <queue>
#include <algorithm>
#include <unordered_map>
#include <math.h>
#include<cstdio>
#include<iostream>
#include<string.h>
#include<map>
using namespace std;
const int MAX = 11e3+5;
char ch[MAX];
int cnt[50];

int main(){
    int n;
    scanf("%d",&n);
    priority_queue<int,vector<int>, greater<int>> q;//创建优先队列，降序排列，小根堆
    while(n--){
        int m, len, mm;
        scanf("%d%s",&m,&ch);
        len = strlen(ch);

        for(int i = 0;i<len;++i){//统计字符串中字母出现的频率
            cnt[ch[i]-'a']++;
        }

        for(int i = 0;i<26;++i){//将统计的字母出现的次数放进队列中
            if(cnt[i]) q.push(cnt[i]);
        }

        int wpl = 0;
        while(q.size()>1){
            int min1 = q.top();
            q.pop();

            int min2 = q.top();
            q.pop();

            wpl += min1 + min2;
            q.push(min1 + min2);
        }

        if(wpl==0) wpl = q.top();
        if(wpl<=m) printf("yes\n");
        else printf("no\n");

        while (!q.empty())
        {
            q.pop();
        }

        memset(cnt, 0, sizeof(cnt));//重置
    }

    return 0;
}

参考：Safe Or Unsafe — HDU - 2527 (哈夫曼树 + 优先队列)