huffman编码

1. #include<iostream>
2. #include<stdlib.h>
3. #include<string>
4. using namespace std;
6. #define MAX 1000
7. typedef char ElemType;
8. typedef int PerType;
9. typedef struct Elem{
10.       ElemType data;
11.       PerType percent;
12.       string value;
13.         }Elem;
14. typedef struct {
15.         PerType weight;
16.         int parent,lchild,rchild;
17.         }HTNode,*HuffmanTree;
19. void Input(Elem *(&a),int n)
20. {
21.    if(!(a=new Elem[n])) exit(1);
22.    for(int i=0;i<n;i++)
23.    cin>>a[i].data>>a[i].percent;  }
25. void Huffman(HuffmanTree &h,Elem *a,int n)
26. {
27.     int m=2*n-1,i=0,j,m1,m2,x1,x2;
28.    if(!(h=new HTNode[m])) exit(1);
29.  for(;i<n;i++)
30.        {
31.             h[i].weight=a[i].percent;
32.             h[i].parent=h[i].lchild=h[i].rchild=-1;}
33.     for(;i<m;i++)
34.        {
35.             h[i].weight=0;
36.             h[i].parent=h[i].lchild=h[i].rchild=-1;}
37.     for(i=0;i<n-1;i++)
38.        {
39.            m1=m2=MAX;
40.            x1=x2=0;
41.            for(j=0;j<n+i;j++)
42.               if(h[j].parent==-1&&h[j].weight<m1)
43.                  {
44.                     m2=m1;x2=x1;
45.                     m1=h[j].weight; 
46.                     x1=j;}      
47.               else if(h[j].parent==-1&&h[j].weight<m2)
48.                 {
49.                     m2=h[j].weight;
50.                     x2=j; }
51.     if(x1>x2) x1^=x2^=x1^=x2;
52.           h[x1].parent=h[x2].parent=n+i;
53.           h[n+i].weight=h[x1].weight+h[x2].weight;
54.           h[n+i].lchild=x1;
55.           h[n+i].rchild=x2;}
56.     
57.     int c,f;
58.     for(i=0;i<n;i++)
59.     {  
60.        string s;
61.        for(c=i,f=h[i].parent;f!=-1;c=f,f=h[f].parent)
62.          {
63.              if(h[f].lchild==c) s="0"+s;
64.              else s="1"+s;}
65.        a[i].value=s; }
66. }
68. void print(Elem *a,int n)
69. {
70. for(int i=0;i<n;i++)
71. cout<<a[i].data<<" "<<a[i].value<<endl;
72. cout<<endl;}
73. string Match(Elem *a,ElemType c,int n)
74. {
75.   int i=0;
76.   for(;i<n;i++)
77.     if(a[i].data==c)   return a[i].value;
78.   if(i==n)exit(1); }      
79.        
80.        
81. int main()
82. {
83.     int n;
84.     Elem *a;
85.     string s;
86.     HuffmanTree h;
87.     cin>>n;
88.     Input(a,n);
89.     Huffman(h,a,n);
90.  print(a,n);
91.  while(cin>>s)
92.     {for(size_t i=0;i<s.size();i++)
93.   cout<<Match(a,s[i],n);
94.  cout<<endl;}
95.     return 0;
96.     }