Cod sursa(job #968338)

Utilizator dropsdrop source drops Data 1 iulie 2013 23:05:49
Problema Coduri Huffman Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.33 kb
#include <iostream>
#include <fstream>
#include <cmath>
#include <vector>
#include <bitset>
#include <queue>
#include <deque>
#include <list>
#include <set>
#include <ctime>
#include <string>
#include <cstring>
#include <iomanip>
#include <algorithm>
using namespace std;
ifstream ff("huffman.in");
ofstream gg("huffman.out");
#define max 1000013

struct huf{ long long v; int k; struct huf *s,*d; 
	huf(long long v0=0,int k0=0){v=v0;k=k0;s=d=0;}
	huf(long long v0, huf *s0, huf *d0){v=v0;s=s0;d=d0;k=0;} }*h,
	*qq[2][max];
int n, ll[max];
long long vv[max];

long long dfs(huf *r,int k,long long x){
	long long l=0;
	if(r->s==0 && r->d==0){
		vv[r->k]=x;
		ll[r->k]=k;
		return k*r->v;
	}
	if(r->s!=0) l+=dfs(r->s,k+1,2*x);  
	if(r->d!=0) l+=dfs(r->d,k+1,2*x+1);
	return l;
}

void arb(){
	huf *s,*d;
	int p, u, i;
	p=1; u=0; i=0;
	while((i<n || p<=u)){
		if(i==n || (p<=u && qq[1][p]->v<qq[0][i]->v))s=qq[1][p++]; else
			s=qq[0][i++];
		if(i==n || (p<=u && qq[1][p]->v<qq[0][i]->v))d=qq[1][p++]; else
			d=qq[0][i++];
			if(s==0||d==0){ break; }
			qq[1][++u]=new huf(s->v+d->v, s, d);
	}
	h=qq[1][u];
	//gg << dfs(h,0,0) << "\n";
	//for(int i=0;i<n;i++) gg << ll[i] << " " << vv[i] << "\n";
}

int main(){
	ff >> n;
	for(int i=0;i<n;i++){
		ff >> vv[i];
		qq[0][i]=new huf(vv[i],i);
	}
	arb();
}