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

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

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

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

#include <cstdio>                                     /************/
#include <cstring>                                    /*    by    */
#include <iostream>                                   /*   kkke   */
#include <algorithm>                                  /************/
#include <queue>                                                       //2016.1.24
#include <conio.h>

#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<tree_size;i++)fread(&tree[i],sizeof(tree[i]),1,infp);
	fread(&bit_cnt,sizeof(bit_cnt),1,infp);
	
	BYTE a1;
	int a2=8;
	int nown=1;
	int b=0;
	system("cls");
	printf("正在解压文件... %3d.%02d%%",b/100,b%100);
	for(long long i=1;i<=bit_cnt;i++)
	{
		if(a2==8)
		{
			a2=0;
		    fread(&a1,sizeof(a1),1,infp);
		}
		if(a1&((BYTE)1<<a2))nown=tree[nown].son[1];
		else nown=tree[nown].son[0];
		if(nown>=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<tree_size;i++)fwrite(&tree[i],sizeof(tree[i]),1,outfp);
	bit_cnt=0ULL;
	for(int i=0;i<MAX_BYTE;i++)bit_cnt+=(ULL)hehe[i]*(ULL)cnt[i+MAX_BYTE];
	fwrite(&bit_cnt,sizeof(bit_cnt),1,outfp);
	
	int b=0;
	system("cls");
	printf("生成文件总大小: %.3f KB\n",((double)bit_cnt+8.0*(double)sizeof(tree))/8192.0);
	printf("压缩率: %.3f%%\n",((double)bit_cnt+8.0*(double)sizeof(tree))/0.08/(double)filesize);
	printf("正在生成压缩文件... %3d.%02d%%",b/100,b%100);
	
	BYTE a;
	BYTE a1=0;
	int a2=0;
	for(int i=1;i<=filesize;i++)
	{
		fread(&a,sizeof(a),1,infp);
		for(int j=0;j<hehe[a];j++)
		{
			if(kkke[a][j])a1|=((BYTE)1<<a2);
			a2++;
			if(a2==8)
			{
				a2=0;
				fwrite(&a1,sizeof(a1),1,outfp);
				a1=0;
			}
		}
		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);
	}
	putchar('\n');
	if(a2)fwrite(&a1,sizeof(a1),1,outfp);
	if(fclose(infp))
	{
		printf("关闭输入文件 %s 失败\n",in_file_name);
		exit(1);
	}
	if(fclose(outfp))
	{
		printf("关闭输出文件 %s 失败\n",out_file_name);
		exit(1);
	}
}

void dfs(int k,int b)
{
	if(k>=MAX_BYTE)
	{
		k-=MAX_BYTE;
		for(int i=0;i<b;i++)kkke[k][i]=kkke1[i];
		hehe[k]=b;
	}
	else
	{
		kkke1[b]=0;
		dfs(tree[k].son[0],b+1);
		kkke1[b]=1;
		dfs(tree[k].son[1],b+1);
	}
}

void build_tree()
{
    system("cls");
	printf("正在分析文件...\n");
	priority_queue<int,vector<int>,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);
	}
}