Cod sursa(job #668613)

Utilizator eddieOlariu Eduard Iuliu eddie Data 25 ianuarie 2012 10:53:03
Problema Coduri Huffman Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.36 kb


#include <stdio.h>

 

using namespace std;

 

#define maxn 1000100
#define inf 1LL * 1000000000 * 1000000000

 
class Huffman{
	long m, j, k, p, l[2], r[2],i;
	void df(long, long long, long);
public:
	struct nod
	{
		long long v;
		long f[2];
	} nod[2*maxn];
 
	long n,q[2][maxn];
	long  lg[maxn];
	long long b[maxn], sol;
	
	void solve();
};
 
void Huffman::df(long poz, long long cod, long nivel)
{
	if(nod[poz].f[0])
	{
		df(nod[poz].f[0], cod*2, nivel+1);
		df(nod[poz].f[1], cod*2+1, nivel+1);
		return;
	}
	lg[poz]=nivel;
	b[poz]=cod;
}
 
void Huffman::solve()
{
	k=n;
	l[0]=l[1]=1;
	r[0]=n;
	while(l[0]+l[1]<=r[0]+r[1])
	{
		++k;
		for(j=0; j<2; j++)
		{
			p=0;
			m=inf;
			for(i=0; i<2; i++)
			{
				if(nod[q[i][l[i]]].v<m && l[i]<=r[i])
				{	
					p=i;
					m=nod[q[i][l[i]]].v;
				}
			}
			nod[k].f[j]=q[p][l[p]];
			nod[k].v+=nod[q[p][l[p]]].v;	
			l[p]++;
		}
	sol+=nod[k].v;
	q[1][++r[1]]=k;
	}
	df(k, 0, 0);
}            
 
int main()
{
	freopen("huffman.in", "r", stdin);
	freopen("huffman.out", "w", stdout);
	Huffman code;
	int i;
	scanf("%d", &code.n);
	for(i=1; i<=code.n; i++)
	{
		scanf("%d", &code.nod[i].v);
		code.q[0][i]=i;
	}
	code.solve();
	printf("%lld\n", code.sol);
	
	for(i=1; i<=code.n; i++)
	{
	printf("%d %lld\n", code.lg[i], code.b[i]);
	}
	return 0;
}