PTA Huffman树

PTA 哈夫曼编码

给定一段文字，如果我们统计出字母出现的频率，是可以根据哈夫曼算法给出一套编码，使得用此编码压缩原文可以得到最短的编码总长。然而哈夫曼编码并不是唯一的。例如对字符串"aaaxuaxz"，容易得到字母 ‘a’、‘x’、‘u’、‘z’ 的出现频率对应为 4、2、1、1。我们可以设计编码 {‘a’=0, ‘x’=10, ‘u’=110, ‘z’=111}，也可以用另一套 {‘a’=1, ‘x’=01, ‘u’=001, ‘z’=000}，还可以用 {‘a’=0, ‘x’=11, ‘u’=100, ‘z’=101}，三套编码都可以把原文压缩到 14 个字节。但是 {‘a’=0, ‘x’=01, ‘u’=011, ‘z’=001} 就不是哈夫曼编码，因为用这套编码压缩得到 00001011001001 后，解码的结果不唯一，“aaaxuaxz” 和 “aazuaxax” 都可以对应解码的结果。本题就请你判断任一套编码是否哈夫曼编码。

输入格式：
首先第一行给出一个正整数 N（2≤N≤63），随后第二行给出 N 个不重复的字符及其出现频率，格式如下：c[1] f[1] c[2] f[2] … c[N] f[N]

其中c[i]是集合{‘0’ - ‘9’, ‘a’ - ‘z’, ‘A’ - ‘Z’, ‘_’}中的字符；f[i]是c[i]的出现频率，为不超过 1000 的整数。再下一行给出一个正整数 M（≤1000），随后是 M 套待检的编码。每套编码占 N 行，格式为：c[i] code[i]
其中c[i]是第i个字符；code[i]是不超过63个’0’和’1’的非空字符串。

输出格式：
对每套待检编码，如果是正确的哈夫曼编码，就在一行中输出"Yes"，否则输出"No"。

注意：最优编码并不一定通过哈夫曼算法得到。任何能压缩到最优长度的前缀编码都应被判为正确。

输入样例：
7
A 1 B 1 C 1 D 3 E 3 F 6 G 6
4
A 00000
B 00001
C 0001
D 001
E 01
F 10
G 11
A 01010
B 01011
C 0100
D 011
E 10
F 11
G 00
A 000
B 001
C 010
D 011
E 100
F 101
G 110
A 00000
B 00001
C 0001
D 001
E 00
F 10
G 11

输出样例：
Yes
Yes
No
No

看着挺复杂的，写起来也挺复杂，但是不难，细心慢慢写，慢慢调就应该可以。
首先要理解Huffman树，就是最优权值，离散数学讲过这一部分，数学要好好学啊。就是找目前权值最小的两个结点合成一个新节点，再继续这个操作就可以。
这个题要构造Huffman树然后再判断输入的是否是Huffman树，有可能权值最小，但是不符合Huffman树的规则，即所有编码不应该是其他结点编码的前缀。

如果真的写不下来就找个有时间或者很无聊的时候再复现一下。
加油！

代码：

#include<bits/stdc++.h>
using namespace std;

typedef struct
{
    int weight;//权重
    int parent;//亲结点
} HTNode,*HuffmanTree;


void select(HuffmanTree &HT,int n,int &s1,int &s2)
{
    int minum=0;//哈夫曼树中最小的权值d jie idan
    //找到第一个没有双亲结点的结点，与其他结点的权值作比较

    for(int i=1; i<=n; i++)
    {

        if(HT[i].parent==0)
        {
            {
                minum=i;
                break;
            }
        }
    }
    for(int i=1; i<=n; i++)
    {
        if(HT[i].parent==0)
            if(HT[i].weight<HT[minum].weight)
                minum=i;//找到权值最小而且没有亲结点的结点
    }
    s1=minum;
    //找第二个值
    for(int i=1;i<=n;i++)
    {
        if(HT[i].parent==0&&i!=s1)
        {minum=i;break;}
    }
    for(int i=1; i<=n; i++)
    {
        if(HT[i].parent==0&&i!=s1)
        {
            if(HT[i].weight<HT[minum].weight)
                minum=i;
        }
    }
    s2=minum;





}

void HuffmanCoding(HuffmanTree &hf,int *a,int n)
{

    int m=2*n-1;//有2n-1个节点
    int min1,min2;
    hf=(HuffmanTree)malloc((m+1)*sizeof(HTNode));
    for(int i=1; i<=n; i++) //初始化原来存在的n个点
    {
        hf[i].weight=a[i];
        hf[i].parent=0;

    }
    for(int i=n+1; i<=m; i++) //初始化2*n-1剩余的点
    {
        hf[i].weight=0;
        hf[i].parent=0;
    }
    for(int i=n+1; i<=m; i++) //创建Huffman树
    {
        select(hf,i-1,min1,min2);//选当前没有父结点的两个权值最小的结点形成一个小树
        hf[i].weight=hf[min1].weight+hf[min2].weight;
        hf[min1].parent=i;
        hf[min2].parent=i;
    }



}



int main()
{
    HuffmanTree hf;
    int m,n;
    char c[1003];
    int a[1003];

    cin>>m;

    for(int i=0; i<m; i++)
    {
        cin>>c[i]>>a[i+1];//输入每个结点及其权值
    }
    HuffmanCoding(hf,a,m);//Huffman树创建完成

    //验证
    cin>>n;
    int WPL = 0;//初始权值设为0

    for(int i=1; i<2*m-1; i++)
    {
        WPL+=hf[i].weight;
    }

    for(int i=0; i<n; i++)
    {
        int sum=0;
        string code[1003]="";
        for(int j=0; j<m; j++)
        {
            cin>>c[j]>>code[j];
            sum=sum+(a[j+1]*code[j].length());//求权值
        }
        if(sum!=WPL)
        {
            //cout<<WPL<<"&"<<sum;
            cout<<"No"<<endl;

        }//权值不是最小，直接排除
        //权值符合，看是否编码规则符合Huffman树编码规则；

        else
        {
            int flag=0;
            for(int i=0; i<m; i++)
            {
                int len=code[i].length();
                for(int j=0; j<m; j++)
                {
                    int len_=code[j].length();
                    if(len<=len_&&i!=j)
                    {
                        if(code[i]==code[j].substr(0,len))
                            flag=1;
                        break;
                    }


                }
                if(flag)
                    break;

            }


            if(flag)
                cout<<"No"<<endl;
            else
                cout<<"Yes"<<endl;
        }


    }







}