用哈弗曼编码实现文件压缩和解压(改进集成版1.0)

    将压缩和解压放在了一个程序里面，并加入了进度显示功能。

    并不完善，因为使用'\b'回退符来达到进度的数字能够变化，可是在需要操作的文件较小时会闪动比较严重，还会消耗多余的资源在显示上面，所以执行效率并不算高，可能在学会其他知识，比如图形化界面时我会再来改进它。

    其他内容在旧版中有写，说多了都是泪 :http://blog.csdn.net/tookkke/article/details/50529838

#include                                      /************/
#include                                     /*    by    */
#include                                    /*   kkke   */
#include                                   /************/
#include                                                        //2016.1.24
#include 

#define MAX_WORD (65536)
#define MAX_BYTE (256)

using namespace std;

typedef unsigned char BYTE;
typedef unsigned short WORD;
typedef unsigned long DWORD;
typedef unsigned long long ULL;

char in_file_name[80];
char out_file_name[80];
int cnt[MAX_BYTE*2];//number of each BYTE
int filesize;
ULL bit_cnt;
int kkke[MAX_BYTE][MAX_BYTE];
int kkke1[MAX_BYTE];
int hehe[MAX_BYTE];

const int tree_size=MAX_BYTE*2;
struct the_tree{
	int son[2];
}tree[MAX_BYTE*2];//root is 1

struct cmp{
	bool operator()(int a,int b)
	{
		return cnt[a]>cnt[b];
	}
};

void read_data();
void build_tree();
void dfs(int k,int b);
void output();

void kdecompress()
{
	system("cls");
	printf("解压\n");
	printf("请输入待解压文件名(不加.kcps):");
	scanf("%s",in_file_name);
	printf("请选择 输出文件名(会覆盖重名文件):\n");
	printf("    1. %s\n",in_file_name);
	printf("    2. 手动输入\n");
	char c;
	while((c=getch())!='1'&&c!='2');
	if(c=='1')strcpy(out_file_name,in_file_name);
	else
	{
		printf("请输入输出文件名:");
		scanf("%s",out_file_name);
	}
	strcat(in_file_name,".kcps");
	printf("将会生成 %s\n",out_file_name);
	printf("    取消： ESC\n");
	printf("    确认： 回车\n");
	while((c=getch())!=13&&c!=27);
	if(c==27)return;
	
	FILE *infp=fopen(in_file_name,"rb");
	if(infp==NULL)
	{
		printf("待解压文件 %s 打开失败\n",in_file_name);
		exit(1);
	}
	FILE *outfp=fopen(out_file_name,"wb");
	if(outfp==NULL)
	{
		printf("解压文件 %s 创建失败\n",out_file_name);
		exit(1);
	}
	
	for(int i=1;i=MAX_BYTE)
		{
			BYTE aa=nown-MAX_BYTE;
			fwrite(&aa,sizeof(aa),1,outfp);
			nown=1;
		}
		a2++;
		if(b==i*10000/bit_cnt)continue;
	    b=i*10000/bit_cnt;
	    for(int k=0;k<7;k++)putchar('\b');
	    printf("%3d.%02d%%",b/100,b%100);
	}
	putchar('\n');
	if(fclose(infp))
	{
		printf("关闭输入文件 %s 失败\n",in_file_name);
		exit(1);
	}
	if(fclose(outfp))
	{
		printf("关闭输出文件 %s 失败\n",out_file_name);
		exit(1);
	}
	system("pause");
}

void kcompress()
{
	system("cls");
	printf("压缩\n");
	printf("请输入待压缩文件名:");
	scanf("%s",in_file_name);
	printf("请选择 输出文件名(会覆盖重名文件):\n");
	printf("    1. %s.kcps\n",in_file_name);
	printf("    2. 手动输入(.kcps)\n");
	char c;
	while((c=getch())!='1'&&c!='2');
	if(c=='1')strcpy(out_file_name,in_file_name);
	else
	{
		printf("请输入输出文件名:");
		scanf("%s",out_file_name);
	}
	strcat(out_file_name,".kcps");
	printf("将会生成 %s\n",out_file_name);
	printf("    取消： ESC\n");
	printf("    确认： 回车\n");
	while((c=getch())!=13&&c!=27);
	if(c==27)return;
	
	read_data();
	build_tree();
	dfs(1,0);
	output();
	system("pause");
}

int main()
{
	while(true)
	{
		system("cls");
		printf("@kkke随便写写的压缩程序：\n");
		printf("请置于待操作文件同一文件夹\n");
		printf("选择    1. 压缩\n");
		printf("        2. 解压\n");
		char c;
		while((c=getch())!='1'&&c!='2');
		if(c=='1')
		{
			printf("已选择 压缩\n");
			printf("    取消： ESC\n");
			printf("    确认： 回车\n");
			while((c=getch())!=13&&c!=27);
			if(c==27)continue;
		    kcompress();
		}
		else
		{
			printf("已选择 解压\n");
			printf("    取消： ESC\n");
			printf("    确认： 回车\n");
			while((c=getch())!=13&&c!=27);
			if(c==27)continue;
		    kdecompress();
		}
	}
	return 0;
}

void output()
{
	FILE *infp=fopen(in_file_name,"rb");
	if(infp==NULL)
	{
		printf("待压缩文件 %s 打开失败\n",in_file_name);
		exit(1);
	}
	FILE *outfp=fopen(out_file_name,"wb");
	if(outfp==NULL)
	{
		printf("压缩文件 %s 创建失败\n",out_file_name);
		exit(1);
	}
	
	for(int i=1;i=MAX_BYTE)
	{
		k-=MAX_BYTE;
		for(int i=0;i,cmp>q;
	for(int i=MAX_BYTE+MAX_BYTE-1;i>=MAX_BYTE;i--)q.push(i);
	for(int i=MAX_BYTE-1;i;i--)
	{
		tree[i].son[0]=q.top();q.pop();
		tree[i].son[1]=q.top();q.pop(); 
		cnt[i]=cnt[tree[i].son[0]]+cnt[tree[i].son[1]];
		q.push(i);
	}
}

void read_data()
{
	memset(cnt,0,sizeof(cnt));
	FILE *infp=fopen(in_file_name,"rb");
	if(infp==NULL)
	{
		printf("未找到待压缩文件\n");
		exit(1);
	}
    fseek(infp,0,SEEK_END);
	filesize=ftell(infp);
	fseek(infp,0,SEEK_SET);
	BYTE a;
	int b=0;
	system("cls");
	printf("文件总大小: %.3f KB\n",(double)filesize/1024.0);
	printf("正在读取文件... %3d.%02d%%",b/100,b%100);
	for(int i=1;i<=filesize;i++)
	{
	    fread(&a,sizeof(a),1,infp);
	    cnt[a+MAX_BYTE]++;
	    if(b==(ULL)i*10000/filesize)continue;
	    b=(ULL)i*10000/filesize;
	    for(int k=0;k<7;k++)putchar('\b');
		printf("%3d.%02d%%",b/100,b%100);
	}
	if(fclose(infp))
	{
		printf("关闭输入文件失败\n");
		exit(1);
	}
}