7-4 哈夫曼编码 (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

#include <iostream>
using namespace std;

struct ElemType
{
  int weight;                  //假定权值为整数
  int parent, lchild, rchild;       //游标
};

class HuffmanTree
{
public:
	HuffmanTree(int w[ ], int n);
	HuffmanTree( );
	int Print( );
private:
	ElemType *huffTree;
	int num;
	void Select(int n, int &i1, int &i2);
};

void HuffmanTree :: Select(int n, int &i1, int &i2)
{
	int i = 0, temp;
	for ( ; i < n; i++)
		if (huffTree[i].parent == -1) {i1 = i; break;}
	for (i = i + 1; i < n; i++)
		if (huffTree[i].parent == -1) {i2 = i;break;}
	if (huffTree[i1].weight > huffTree[i2].weight)
	{
		temp = i1; i1 = i2;i2 = temp;
	}
	for (i = i + 1; i < n; i++)
	{
		if (huffTree[i].parent == -1)
		{
			if (huffTree[i].weight < huffTree[i1].weight)
			{
				i2 = i1; i1 = i;
			}
			else if (huffTree[i].weight < huffTree[i2].weight)
			{
				i2 = i;
			}
		}
	}
}

HuffmanTree :: HuffmanTree(int w[ ], int n)
{
	int i, k, i1, i2;
	huffTree = new ElemType [2*n-1];
	num = n;
	for (i = 0; i < 2*num-1; i++)         //初始化，所有结点均没有双亲和孩子
	{
		huffTree[i].parent = -1;
		huffTree[i].lchild = huffTree[i].rchild = -1;
 	}
  	for (i = 0; i < num; i++)             //存储叶子结点的权值
    	huffTree[i].weight = w[i];
  	for (k = num; k < 2*num-1; k++)        //n-1次合并
  	{
    	Select(k, i1, i2);        //权值最小的根结点下标为i1和i2
		huffTree[k].weight = huffTree[i1].weight + huffTree[i2].weight;
    	huffTree[i1].parent = k; huffTree[i2].parent = k;
    	huffTree[k].lchild = i1; huffTree[k].rchild = i2;
  	}
}

int HuffmanTree :: Print( )
{
    int i, k;
    int sum=0;
    //cout << "每个叶子到根结点的路径是:" << endl;
    for (i = 0; i < num; i++)
    {
        //cout << huffTree[i].weight;
        int cnt=0;
        k = huffTree[i].parent;
        while (k != -1)
        {
            cnt++;
            //cout << "-->" << huffTree[k].weight;
            k = huffTree[k].parent;
        }
        sum+=cnt*huffTree[i].weight;
        //cout <<cnt*huffTree[i].weight<< endl;
    }
    return sum;
}
int main( )
{
    int n,w[100],i,t;
    cin>>n;
    string s[100];
    char a;
    for(i=0; i<n; i++)
    {
        cin>>a>>w[i];
    }
    HuffmanTree T{w, n};
    int sum=T.Print();
    cin>>t;
    while(t--)
    {
        int sum2=0;
        for(i=0; i<n; i++)
        {
            cin>>a>>s[i];
            sum2+=s[i].length()*w[i];
        }
        if(sum2!=sum)
        {
            cout<<"No"<<endl;
        }
        else
        {
            int flag=0,j;
            for(i=0; i<n; i++)
            {
                for(j=i+1; j<n; j++)
                {
                    if(s[i].length()==s[j].length())
                    {
                        if(s[i]==s[j])
                        {
                            flag=1;
                        }
                    }
                    else
                    {
                        int flag1=1;
                        for(int k=0; k<min(s[i].length(),s[j].length()); k++)
                        {
                            if(s[i][k]!=s[j][k])
                            {
                                flag1=0;
                                break;
                            }
                        }
                        if(flag1==1)
                        {
                            flag=1;
                            break;
                        }
                    }
                }
                if(flag==1){
                    break;
                }
            }
            if(flag==0){
                cout<<"Yes"<<endl;
            }else{
                cout<<"No"<<endl;
            }
        }
    }
    return 0;

}