hdu 题目1053Entropy (赫夫曼树)

 

Entropy

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 3152    Accepted Submission(s): 1209


Problem Description
An entropy encoder is a data encoding method that achieves lossless data compression by encoding a message with “wasted” or “extra” information removed. In other words, entropy encoding removes information that was not necessary in the first place to accurately encode the message. A high degree of entropy implies a message with a great deal of wasted information; english text encoded in ASCII is an example of a message type that has very high entropy. Already compressed messages, such as JPEG graphics or ZIP archives, have very little entropy and do not benefit from further attempts at entropy encoding.

English text encoded in ASCII has a high degree of entropy because all characters are encoded using the same number of bits, eight. It is a known fact that the letters E, L, N, R, S and T occur at a considerably higher frequency than do most other letters in english text. If a way could be found to encode just these letters with four bits, then the new encoding would be smaller, would contain all the original information, and would have less entropy. ASCII uses a fixed number of bits for a reason, however: it’s easy, since one is always dealing with a fixed number of bits to represent each possible glyph or character. How would an encoding scheme that used four bits for the above letters be able to distinguish between the four-bit codes and eight-bit codes? This seemingly difficult problem is solved using what is known as a “prefix-free variable-length” encoding.

In such an encoding, any number of bits can be used to represent any glyph, and glyphs not present in the message are simply not encoded. However, in order to be able to recover the information, no bit pattern that encodes a glyph is allowed to be the prefix of any other encoding bit pattern. This allows the encoded bitstream to be read bit by bit, and whenever a set of bits is encountered that represents a glyph, that glyph can be decoded. If the prefix-free constraint was not enforced, then such a decoding would be impossible.

Consider the text “AAAAABCD”. Using ASCII, encoding this would require 64 bits. If, instead, we encode “A” with the bit pattern “00”, “B” with “01”, “C” with “10”, and “D” with “11” then we can encode this text in only 16 bits; the resulting bit pattern would be “0000000000011011”. This is still a fixed-length encoding, however; we’re using two bits per glyph instead of eight. Since the glyph “A” occurs with greater frequency, could we do better by encoding it with fewer bits? In fact we can, but in order to maintain a prefix-free encoding, some of the other bit patterns will become longer than two bits. An optimal encoding is to encode “A” with “0”, “B” with “10”, “C” with “110”, and “D” with “111”. (This is clearly not the only optimal encoding, as it is obvious that the encodings for B, C and D could be interchanged freely for any given encoding without increasing the size of the final encoded message.) Using this encoding, the message encodes in only 13 bits to “0000010110111”, a compression ratio of 4.9 to 1 (that is, each bit in the final encoded message represents as much information as did 4.9 bits in the original encoding). Read through this bit pattern from left to right and you’ll see that the prefix-free encoding makes it simple to decode this into the original text even though the codes have varying bit lengths.

As a second example, consider the text “THE CAT IN THE HAT”. In this text, the letter “T” and the space character both occur with the highest frequency, so they will clearly have the shortest encoding bit patterns in an optimal encoding. The letters “C”, “I’ and “N” only occur once, however, so they will have the longest codes.

There are many possible sets of prefix-free variable-length bit patterns that would yield the optimal encoding, that is, that would allow the text to be encoded in the fewest number of bits. One such optimal encoding is to encode spaces with “00”, “A” with “100”, “C” with “1110”, “E” with “1111”, “H” with “110”, “I” with “1010”, “N” with “1011” and “T” with “01”. The optimal encoding therefore requires only 51 bits compared to the 144 that would be necessary to encode the message with 8-bit ASCII encoding, a compression ratio of 2.8 to 1.
 


 

Input
The input file will contain a list of text strings, one per line. The text strings will consist only of uppercase alphanumeric characters and underscores (which are used in place of spaces). The end of the input will be signalled by a line containing only the word “END” as the text string. This line should not be processed.
 


 

Output
For each text string in the input, output the length in bits of the 8-bit ASCII encoding, the length in bits of an optimal prefix-free variable-length encoding, and the compression ratio accurate to one decimal point.
 


 

Sample Input
AAAAABCD THE_CAT_IN_THE_HAT END
 


 

Sample Output
64 13 4.9 144 51 2.8

 

数组tree下标0代表‘_’,1---26代表A---Z,weigth代表出现的次数,deep代表结点所在赫夫曼树的深度,

建立的新节点放在下标27开始,

最后将0---26中元素的权值 (出现次数) weigth * deep 累加起来;

 

 

代码: 请无视注释,

#include <string.h>
#include <stdio.h>
#include <iostream>
#include <stdlib.h>

using namespace std;

typedef struct TREE
{
	int weigth,parent,lchild,rchild,deep;
}TREE;

TREE tree[55];
bool visit[55];


int len_huff(int n)
{
	int j,i,k;
	memset(visit,0,sizeof(visit));
	int min1,min2,index1,index2;
	
	int num = 0;
	for(i=0;i<27;i++)
	{
		if(tree[i].weigth != 0)
		{
			k=i;
			num++;
		} 	
	
	}
		
	if(num==1) return tree[k].weigth;
//	printf("num=%d\n",num);
	
	for(i=27;i<26+num;i++)
	{
		
		
		min1 = min2 = 999999;
		for(j=0;j<i;j++)//寻找最小的两个元素
		{
			if(!visit[j])
			{
				if( tree[j].weigth!=0 && tree[j].weigth<min1)
				{
					min2 = min1;
					index2 = index1;
					min1 = tree[j].weigth;
					index1 = j;		
				}
				else if( tree[j].weigth!=0 && tree[j].weigth<min2)
				{
					index2 = j;
					min2 = tree[j].weigth;
				}										
			}//if 							
		}	
		visit[index1] = visit[index2] = 1;
//		printf("index1=%d,min1=%d,index2=%d,min2=%d\n",index1,min1,index2,min2);//找到最小的两个 
			
	//	tree[index1].deep++;
	//	tree[index2].deep++;
		
		tree[i].weigth = min1 + min2;  //新结点
		tree[i].lchild = index1; 
		tree[i].rchild = index2;
		tree[index1].parent = tree[index2].parent = i;
	}
	
//	for(i=27;i<55;i++)
//	{
//		int child = tree[i].lchild;
//		int parent,dp=1;
////				printf("lchild=%d\n",child);
//		while(child!=-1)
//		{
//			dp++;
//			parent = child;
//			child = tree[child].lchild;
//		}	
//		tree[parent].deep = dp;
////		
//		printf("pa=%d,de=%d\n",parent,dp);
//		child = tree[i].rchild;
//		dp=1;
//		while(child!=-1)
//		{
//			dp++;
//			parent = child;
//			child = tree[child].rchild;
//		}	
//		tree[parent].deep = dp;
//	//	printf("rchild=%d,de=%d\n",child,dp);
//	}



	int len = 0;
	for(i=0;i<27;i++)//计算每个节点的深度
	{
		int parent = tree[i].parent;
		int child ,dp=0;
//				printf("lchild=%d\n",child);
		while(parent !=-1)//不是 根节点 
		{
			dp++;
			child= parent ;
			parent  = tree[parent].parent;
		}	
		tree[i].deep = dp;
		len += tree[i].deep*tree[i].weigth;	
//		
	}
//	printf("%d ",len);
	return len;
					
}
/*
AA
AAAAABCD
THE_CAT_IN_THE_HAT
END
*/
int main()
{
	char s[10001];
	int i;
	while(gets(s))
	{
		if(strcmp(s,"END")==0) break;
	//	printf("%d ",);
		
		memset(tree,-1,sizeof(tree));
		for(i=0;i<55;i++) tree[i].weigth=tree[i].deep=0;
		
		for(i=0;i<strlen(s);i++)
		{
			if(s[i]=='_') tree[0].weigth++;
			else tree[s[i]-'A'+1].weigth++;	
		}
		
		int len2 = len_huff(strlen(s));
		
		int len1 = strlen(s)*8; 
		
		
		double radio = 1.0*len1/len2;
//	printf("len1=%d,len2=%lld,radio=%lf\n",len1,len2,radio);
		printf("%d %d %.1lf\n",strlen(s)*8,len2,radio);
//		printf("    i    weigth   parent    lchild    rchild    deep\n");
//		for(i=0;i<55;i++)
//			printf(" %4d:   %5d    %5d    %5d    %5d    %5d\n",i,tree[i].weigth,tree[i].parent,tree[i].lchild,tree[i].rchild,tree[i].deep);	
		
		
	//	printf("%d ",length_Haffuman(s));
	//	printf(".1lf\n",1.0*strlen(s)/length_Haffuman(s));			
	}
	return 0;	
}