Safe Or Unsafe（简单的霍夫曼编码 优先队列）

Safe Or Unsafe Time Limit:1000MS    Memory Limit:32768KB    64bit IO Format:%I64d & %I64u

Submit Status

Description

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

 

Input

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

 

Output

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

 

Sample Input

    

      2
12
helloworld
66
ithinkyoucandoit 
    

 

Sample Output

  

    no
yes 

   
#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;

int main()
{
    int n;
    int i,j;
    char s[1000];
    int str[26];
    priority_queue< int ,vector<int>,greater<int> >q;

    scanf("%d",&n);

    while(n--)
    {
        int sum;
        int ssum=0;
        memset(str,0,sizeof(str));

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

        scanf("%d",&sum);

        scanf("%s",s);

        for(i=0;s[i];i++)
        {
            int k=s[i]-'a';
            str[k]++;
        }

        for(i=0;i<26;i++)
        {
            if(str[i]!=0)
            {
                 //printf("%d\n",str[i]);
             q.push(str[i]);
            }

        }

        if(q.size()==1)
        {
             ssum=q.top();
            q.pop();

        }
        while(q.size()>1)
        {
            int a=q.top();
            q.pop();
            int b=q.top();
            q.pop();
            ssum+=(a+b);
            q.push(a+b);
        }

        if(ssum>sum)
            printf("no\n");
        else printf("yes\n");
    }

return 0;
}