哈夫曼树的构造

代码

需要注意的地方见代码注释~小编累了就不建标题写了（这几天刷了好多到二叉树的递归算法题，脑壳疼疼555，等小编恢复状态了再统一整理出来），后面有时间再排版哈夫曼树这章吧555

#include<bits/stdc++.h>
using namespace std;

int a[1005];//定义一个数组，接受初始权值 
int sum;//定义sum存储每次相加的结果

typedef struct BiTNode{
	int val;
	BiTNode *lchild,*rchild;
}BiTNode,*BiTree;

bool cmp(BiTNode *a,BiTNode *b){
	return a->val > b->val;
}

BiTNode *CreateHuffuman(int n){//n表示输入序列的长度 
	//将序列构建成结点放入数组b中，区别：a数组存的是权值（数字），b数组存的是结点（树结点的所有属性）
	BiTNode *b[1005];
	for(int i=0;i<n;i++){
		BiTNode *p=(BiTNode *)malloc(sizeof(BiTNode));
		p->val=a[i];
		p->lchild=p->rchild=NULL;
		b[i]=p;
	} 
	sort(b,b+n,cmp);//sort函数指定排序规则cmp,将 [first, last)区间内的元素按照指定的顺序排列。
	BiTNode *r=b[0];//如果只有一个元素的话，进不了循环只能返回空，故先初始化为数组第一个元素 
	int l=n;
	while(l!=1){
		//取出最小的两个值
		BiTNode *p=b[l-1];
		BiTNode *q=b[l-2];
		r=new BiTNode;//申请一个结点的空间
		r->val=p->val + q->val;
		sum+=r->val ;//最小带权路径长度刚好就是新建结点的权值之和 
		r->lchild =p;//最小的作为左子树
		r->rchild =q;//第二小的作为右子树
		//将r插入降序数组b中
		int i,j;
		//找到插入的位置 注意：l-2是因为后面有两个元素准备从数组里删掉，再加入一个新元素，且表示的是下标 现在数组里边元素的区间是 0~l-3 
		for(i=0;i<l-2;i++){
			if(r->val >b[i]->val )
			break;//找到该插入的位置，i已经标记好了 
		}
		//将插入位置后面的元素全部后移
		for(j=l-2;j>i;j--){//结束是j=i 
			b[j]=b[j-1];
		} 
		b[j]=r;//插入新结点
		l--;//序列长度-1：因为删去了两个小的，加入一个新的，总的少了一个
	} 
	return r; 
}

void PreOrder(BiTNode *T){
	if(T){
		cout<<T->val<<" ";
		PreOrder(T->lchild);
		PreOrder(T->rchild);
	}
}

int main(){
	int n;
	while(cin>>n){//多轮输入序列的个数
	for(int i=0;i<n;i++){
		cin>>a[i];
	} 
	BiTNode *T=CreateHuffuman(n);
	PreOrder(T);//先序输出构造的哈夫曼树
	cout<<endl;
	cout<<sum<<endl;//输出sum
	sum=0;//每轮结束后将sum置零
    }
    return 0;
}