HDU2527 Safe Or Unsafe 哈夫曼树

本题链接：http://acm.hdu.edu.cn/showproblem.php?pid=2527

此题考查的是哈夫曼树，构造哈夫曼树并计算出最小编码数即可。

做此题的过程中，提交之后竟然TLE，很是郁闷，找了几遍，还是没找出错误，可以肯定算法是不会错的，这方面不会超时，当测试只有一个字符的时候，发现问题了，出现了死循环，因为当只有1个或一种相同的字符的时候，并没有进行构造哈弗曼数的过程，因此在下面的找最小编码数的时候，出现了死循环，这个地方需要特殊计算最小编码数。改过之后，立刻就AC了，真是感慨，不过还挺满意的，自己构造出了哈夫曼树，继续加油！\(^o^)/~

AC代码如下：

#include<iostream>

using namespace std;

#include<string.h>

struct node

{

    int r,l,p,data;

}ha[100];

char s[100000];

int main()

{

    //freopen("d:\\1.txt","r",stdin);

    int n,a[27],b[27],c[27];

    cin>>n;

    while(n--)

    {

        int m1,h,s1=0;

        int i,j,k=0;    

        cin>>m1;

        memset(a,0,sizeof(a));

        memset(b,0,sizeof(b));

        memset(c,0,sizeof(c));

        scanf("%s",s);

        int l=strlen(s);        

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

        {

            int m=s[i]-96;

            a[m]++;

        }

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

        {

            if(a[i])

            b[++k]=a[i];

        }

        if(k!=1)//特别注意要考虑k=1的情况,因为没有进行构成哈夫曼树的操作,所以将会在下面出现死循环

        {

        //构造哈夫曼树

        memset(ha,0,sizeof(ha));

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

         ha[i].data=b[i];

        for(i=1;i<k;i++)//共有k-1个结点需要补充

        {

            int max=10000000;    

            int m1,m2,x1,x2;

            m1=max;//m2存上一个m1的值,m1存最新的值,x2存上一个x1的值,x1存最新的值

            m2=max;//m1,m2存结点的权值,x1,x2分别存结点的左右孩子的位置

            x1=0;

            x2=0;

            for(j=1;j<k+i;j++)

            {

                if(ha[j].data<m1&&ha[j].p==0)

                {

                    m2=m1;

                    m1=ha[j].data;

                    x2=x1;

                    x1=j;

                }

                else if(ha[j].data<m2&&ha[j].p==0)

                {

                    m2=ha[j].data;

                    x2=j;

                }

            }

            ha[k+i].l=x1;

            ha[k+i].r=x2;

            ha[x1].p=k+i;

            ha[x2].p=k+i;

            ha[k+i].data=ha[x1].data+ha[x2].data;

        } 

        //测试输出树

        /*for(i=1;i<2*k;i++)

        {

            cout<<i<<' '<<ha[i].l<<' '<<ha[i].data<<' '<<ha[i].r<<' '<<ha[i].p<<endl;

        }*/

        //求最小编码数

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

        {

            j=i;

            for(;;)

            {

                h=ha[j].p;

                j=h;

                c[i]++;

                if(h==2*k-1)break;//若不讨论 将会在此死循环 因为h会恒等于0

            }

            s1+=b[i]*c[i];

        }

        }

        else 

        s1=b[1]*1;

        //cout<<"s1="<<s1<<endl;

        

        if(s1<=m1)cout<<"yes"<<endl;

        else cout<<"no"<<endl;

            

    }

    return 0;

}