PTA 7-1 哈夫曼编码 (30分)

PTA 7-1 哈夫曼编码 (30分)

问题描述：

给定一段文字，如果我们统计出字母出现的频率，是可以根据哈夫曼算法给出一套编码，使得用此编码压缩原文可以得到最短的编码总长。然而哈夫曼编码并不是唯一的。例如对字符串"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

解析思路：

判断是否是哈夫曼编码 ——>利用哈夫曼编码的性质：

1. 叶子结点数 n0=n2+1，故哈夫曼树结点总数为 n0+n2=n0+(n0-1)=2*n0-1
2. 前缀编码，任意结点的哈夫曼不能是其余结点哈夫曼编码的前缀

答案：

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

typedef struct tnode
{
    char data;
    struct tnode *lchild,*rchild;
} BiTNode,*BiTree;

int main()
{
    priority_queue<int,vector<int>,greater<int> >Q;
    char data[100];
    int weight[100];
    int n,wpl = 0;
    cin>>n;
    int i,j,k;
    for(i = 0; i < n; i++)
    {
        getchar();
        cin>>data[i]>>weight[i];
        Q.push(weight[i]);
    }

    while(Q.size() > 1)
    {
        int a,b;
        a = Q.top();
        Q.pop();
        b = Q.top();
        Q.pop();
        Q.push(a+b);
        wpl += a+b;
    }
    
    int m;
    cin>>m;
    for(i = 0; i < m; i++)
    {
        int sum2 = 0;
        char ch[100][100];
        for( j = 0; j < n; j++)
        {
            getchar();
            char c;
            cin>>c>>ch[j];
            sum2 += strlen(ch[j])*weight[j];
        }
        printf("sum1 shi %d , sum2 shi %d\n",wpl,sum2);
        if(wpl != sum2)
            printf("No\n");//判断是否满足性一： 叶子结点数 n0=n2+1，故哈夫曼树结点总数为 n0+n2=n0+(n0-1)=2*n0-1
        else
        {
            int flag = 0;
            for(i = 0; i < n; i++)
            {
                for(j = 0; j < n; j++)
                {
                    int x = strlen(ch[i])>strlen(ch[j])?strlen(ch[j]):strlen(ch[i]);
                    for(int k = 0; k < x; k++)
                    {
                        if(ch[i][k] != ch[j][k])
                        {
                            flag = 1;   //判断是否满足性质二：前缀编码，任意结点的哈夫曼不能是其余结点哈夫曼编码的前缀 
                            break;
                        }
                    }
                }
            }
            if(flag)
                printf("Yes\n");
            else
                printf("No\n");
        }
    }
    return 0;
}

可能是编译器的问题还是其他问题，在运行中可能出现小错误，但思路是对的，可以根据思路自行调试改正代码。

谢谢采纳！