main.cpp

huffman树的基本实现. 利用读取TXT文件统计产生huffman树,从而对TXT文件进行压缩和解压.

#include <iostream>
#include <fstream>
using namespace std;
#include "HuffmanTree.h"

const int MaxNum=200;
void main()
{
	bool flag=true;
	int type;
	int n;
	HuffmanTree t1;//声明哈夫曼树;
	while(flag)
	{
		cout<<"请输入要进行的操作:\n1:初始化 2:编码 3:压缩 4:译码 5: 打印代码 \n6:打印编码 7:显示哈夫曼树 8:退出;\nYour Choice:";
		cin>>type;
		switch(type)
		{
	
		case 1:
			t1.Init();
			break;
		
		case 2:
			n=t1.getnumber();
			if(n==0)
			{
				cout<<"Huffman Tree not initialized!\n";
				exit(1);
			}
			t1.Encode(2*n-2);
			break;
		
		case 3:
			t1.Compress();
			break;
		
		case 4:
			t1.Decode();
			break;
		
		case 5:
			t1.PrintHuffmanTreeCode();
			break;
		
		case 6:
			t1.Print();
			break;
		
		case 7:
			t1.TreePrint();
			break;
		
		case 8:
			flag=false;
			break;
		default:
			cout<<"请做出正确选择!\n";
		}
	}
}