【SSL 1409】哈夫曼树(三)【树】

---

解题思路

将输入的每个字母出现的次数当做关键字，构建一个哈夫曼树，然后中序遍历输出字母和哈弗曼编码就好了。

---

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<map>
#define ll long long
using namespace std;

string s;
map<char,int>mp; 
int n;

struct c{
	char c;
	int x,p;
	int ls,rs;
}f[100010],a[100010];

void out(int t,string s)
{
	if(f[t].x==0)return;
	out(f[t].ls,s+'0');
	if(f[t].ls==0&&f[t].rs==0)
		cout<<f[t].c<<":"<<s<<endl;
	out(f[t].rs,s+'1');
}

bool cmp(c l,c r)
{
	if(l.x==r.x)return l.p<r.p;
	return l.x<r.x;
}

int main(){
	cin>>s;
	int len=s.size();
	for(int i=0;i<len;i++)
		mp[s[i]]++;
	for(int i=0;i<len;i++)
	{
		if(mp[s[i]])
		{
			a[++n].x=f[n].x=mp[s[i]];
			a[n].c=f[n].c=s[i];
			a[n].p=n;
			mp[s[i]]=0;
		}	
	}
	int i=n,t=n;
	while(i>1)
	{
		sort(a+1,a+i+1,cmp);
		f[++t].x=a[1].x+a[2].x;
		f[t].ls=a[1].p,f[t].rs=a[2].p;
		a[1].x=f[t].x,a[1].p=t,a[1].c=char('z'+1);
		a[2].x=a[i].x,a[2].p=a[i].p,a[2].c=a[i].c;
		i--;
	}
	out(t,"");
}