HDOJ 2527 Safe Or Unsafe 哈夫曼编码

Safe Or Unsafe

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

Problem 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

/*
HDOJ 2527 哈夫曼编码
计算出Huffman最优编码至少需要的存储位数然后再比较
判断除了叶子结点之外的其他结点之和是否大于题目给出的数字 
1 1 3=> ans=2+5=7
     5
   / \ 
  2  3
 /\  
1 1 

出现好几次wrong
第一次 队列最后还剩一个 !que.empty()错误，
第二次 因为第二次运行队列没清空 还剩一个 所以要出队 
*/ 
#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<queue>
using namespace std;

bool cmp(char a,char b)
{
    return a>b;
}

struct TreeNode{
    
    
    int count;
    
    friend bool operator<(const TreeNode &a,const TreeNode &b)
    {
        return a.count>b.count;
    }
};


int ans;
priority_queue<TreeNode> que;

void haffman()
{
    TreeNode a,b,c;
    while(que.size()>1)//!que.empty()错误1，最后还剩一个 
    {
        
        a=que.top();
        que.pop();
        
        b=que.top();
        que.pop();
        
        c.count=a.count+b.count;
        ans+=c.count;
        
        que.push(c);
    }
    que.pop(); //错误2 队列未清空 影响下次运行 所以这里pop
              //或者 每次重新定义新队列 
}

int main()
{
    char str[1000];
    int t,i,j,len,sum,count;
    
//    freopen("test.txt","r",stdin);
    scanf("%d",&t);
    while(t--)
    {
        
        scanf("%d",&sum);
        scanf("%s",str);
        len=strlen(str);
        sort(str,str+len,cmp);
        
        TreeNode t;
        count=1;
        
        for(i=1;i<len;i++)
        {
            if(str[i]==str[i-1])
                count++;
            else
            {
                t.count=count;
                que.push(t);
                count=1;
            }    
        }
        t.count=count;
        que.push(t);
        ans=0;
        if(que.size()==1)
        {
            if(len<=sum)
                printf("yes\n");
            else
                printf("no\n");
            que.pop();
        }
        else
        {
            haffman();
        //    printf("%d",ans);
            if(ans<=sum)
                printf("yes\n");
            else
                printf("no\n");    
        }
        
    }
    return 0;
}