哈夫曼树模板

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll N=2000000;
int n;
struct tree
{
	int data,rch,lch,num;
}f[N+1],a[N+1];//f数组用来储存哈夫曼树的节点，其中1~n号存的是叶子节点 
//a数组用来给每个二叉树的根节点排序 
bool cmp(tree x,tree y){return x.data<y.data;}
void print(int x,string code)
{
	if(!f[x].lch&&!f[x].rch)
	{
		printf("%d:",f[x].data),cout<<code<<endl;
		return;
	}
	print(f[x].lch,code+"0");
	print(f[x].rch,code+"1");
}
int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&f[i].data);
		a[i].num=i;
		a[i].data=f[i].data;
	}
	for(int i=n,t=n+1;i>1;i--,t++)
	{
		sort(a+1,a+i+1,cmp);
		f[t].data=a[1].data+a[2].data;
		f[t].lch=a[1].num;
		f[t].rch=a[2].num;//构造新的二叉树节点 
		a[1].data=f[t].data;
		a[1].num=t;//将新构造的节点放入排序数组a中 
		a[2].data=a[i].data;
		a[2].num=a[i].num;//将a数组最后一个节点放到前面
	}
	print(2*n-1,"");
	return 0;
}