标程_哈夫曼树(堆优化)

标程_哈夫曼树(堆优化)

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
char ans[2010]="";
struct tree{int s,l,r;}a[2010];//s权值,l左子树地址,r右子树地址
struct jgt{int s;};
bool operator < (const jgt &t1 ,const jgt &t2){return a[t1.s].s>a[t2.s].s;}
priority_queue<jgt>que;
void print(int x,int dep)//中序遍历输出 
{
	if(a[x].l==0&&a[x].r==0)cout<<a[x].s<<':'<<ans<<endl;//叶子节点 
	if(a[x].l>0)ans[dep]='0',ans[dep+1]=0,print(a[x].l,dep+1);//输出左子树
	if(a[x].r>0)ans[dep]='1',ans[dep+1]=0,print(a[x].r,dep+1);//输出右子树 
	return;
}
int main()
{
	int n,m,i;
	jgt tem;
	ios::sync_with_stdio(false);
	for(cin>>n,i=1;i<=n;i++)cin>>a[i].s,a[i].l=a[i].r=0,tem.s=i,que.push(tem);
	for(i=n+1,m=n<<1;i<m;i++)
	{
		a[i].l=que.top().s,que.pop();
		a[i].r=que.top().s,que.pop();
		a[i].s=a[a[i].l].s+a[a[i].r].s;
		tem.s=i,que.push(tem);
	}
	print(m-1,0);
	return 0;
}