HDU2527：Safe Or Unsafe(哈弗曼树)

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

 

一道简单的哈弗曼树，一开始没看懂题意，经过别人告诉我题意，原来只是一道这么简单的哈夫曼树，题目是要求除了叶子节点外的所有节点权值之和，正好数据结构刚刚学了哈夫曼树，趁热打铁。

要注意的是，如果在字符只有一种的情况下，哈夫曼树是建不起来的，要特殊处理

 

#include <stdio.h>

#include <string.h>

#include <algorithm>

using namespace std;



const int L = 1000000+10;

const int inf = 1<<30;

char str[L];

struct kode

{

    int num;

    char c;

} b[L<<2];



struct node

{

    int wei,parent,lson,rson,cover;

    char data;

} a[L<<2];



int main()

{

    int i,j,sum,t,len,lb;

    char ch;

    scanf("%d",&t);

    while(t--)

    {

        scanf("%d%s",&sum,str);

        len = strlen(str);

        sort(str,str+len);

        for(i = 0; i<len<<2; i++)

        {

            a[i].cover = a[i].lson = a[i].rson = a[i].parent = a[i].wei = b[i].num = 0;

            a[i].data = b[i].c = '\0';

        }

        lb = 0;

        b[lb].num = 1;

        b[lb].c = ch = str[0];

        for(i = 1; i<len; i++)

        {

            if(str[i] == ch)

                b[lb].num++;

            else

            {

                lb++;

                ch = str[i];

                b[lb].c = ch;

                b[lb].num = 1;

            }

        }

        if(lb == 0)//只有一种类型的字符，直接比较

        {

            if(b[lb].num<=sum)

                printf("yes\n");

            else

                printf("no\n");

            continue;

        }

        lb++;

        int m = lb*2-1;

        for(i = 0; i<lb; i++)

        {

            a[i].data = b[i].c;

            a[i].wei = b[i].num;

        }

        for(i = 0; i<lb; i++)//建立哈夫曼树

        {

            int m1 = inf,m2 = inf;

            int x = 0,y = 0;

            for(j = 0; j<lb+i; j++)

            {

                if(a[j].wei<m1 && !a[j].cover)

                {

                    m2 = m1;

                    m1 = a[j].wei;

                    y = x;

                    x = j;

                }

                else if(a[j].wei<m2 && !a[j].cover)

                {

                    m2 = a[j].wei;

                    y = j;

                }

            }

            a[x].parent = lb+i;

            a[y].parent = lb+i;

            a[lb+i].lson = x;

            a[lb+i].rson = y;

            a[lb+i].wei = a[x].wei+a[y].wei;

            a[x].cover = a[y].cover = 1;

        }

        int ans = 0;

        for(i = lb; i<2*lb-1; i++)//求除了叶子节点外其他所有节点的权值和

            ans+=a[i].wei;

        if(ans<=sum)

            printf("yes\n");

        else

            printf("no\n");

    }



    return 0;

}