Safe Or Unsafe hdu2572 优先队列 哈弗曼编码

Safe Or Unsafe

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 3120    Accepted Submission(s): 1275

Problem Description

Javac++ 一天在看计算机的书籍的时候，看到了一个有趣的东西！每一串字符都可以被编码成一些数字来储存信息，但是不同的编码方式得到的储存空间是不一样的！并且当储存空间大于一定的值的时候是不安全的！所以Javac++ 就想是否有一种方式是可以得到字符编码最小的空间值！显然这是可以的，因为书上有这一块内容--哈夫曼编码(Huffman Coding)；一个字母的权值等于该字母在字符串中出现的频率。所以Javac++ 想让你帮忙，给你安全数值和一串字符串，并让你判断这个字符串是否是安全的？

 

Input

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

 

Output

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

 

Sample Input

212helloworld66ithinkyoucandoit

 

Sample Output

noyes

哈弗曼编码裸题，用优先队列存储每个次数出现的次数，每次取出靠前的两位，并将和插进去，形成新的节点。

代码

#include<bits/stdc++.h>
using namespace std;
struct point
{
  int a=0;
  friend bool operator < (point s,point w)
  {
      return s.a>w.a;
  }
};
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        point p[26];
        int n,ans=0;
        scanf("%d",&n);
        char s[1000];
        scanf("%s",s);
        int len=strlen(s);
        for(int i=0;i<len;i++)
            p[s[i]-'a'].a++;
        priority_queue<point> q;
        for(int i=0;i<26;i++)
            if(p[i].a>0)
                q.push(p[i]);
        if(q.size()==1)
            ans=len;
        else
        while(q.size()>1)
        {
            int w=q.top().a;
            q.pop();
            int e=q.top().a;
            q.pop();
            ans+=w+e;
            point pp;
            pp.a=w+e;
            q.push(pp);
        }
        if(ans<=n)
        printf("yes\n");
        else
        printf("no\n");
        

    }
    return 0;
}