Cod sursa(job #1222524)

Utilizator Luncasu_VictorVictor Luncasu Luncasu_Victor Data 23 august 2014 15:13:01
Problema Coduri Huffman Scor 95
Compilator cpp Status done
Runda Arhiva educationala Marime 1.92 kb
#include <iostream>
#include <iomanip>
#include <fstream>
#include <cstdio>
#include <cmath>
#include <cstring>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
#define MAX 1000010

struct Node {
	int Freq, Id;
	struct Node *Left, *Right;
	Node() {
		Id = 0;
		Freq = 0;
		Left = Right = 0;
	}
};

struct NodeCompare{
	bool operator() (const Node *A, const Node*B) {
		return A->Freq > B->Freq;
	}
};

vector<Node*> A;
queue<Node*> Q1, Q2;
int N, Len[MAX];
long long Sol, Codes[MAX];

void Huffman() {
	Node *X, *Y, *Z;
	while (Q1.size() + Q2.size() > 1) {
		if (Q1.size() == 0) {
			X = Q2.front();
			Q2.pop();
		} else
		if (Q2.size() == 0) {
			X = Q1.front();
			Q1.pop();
		} else {
			if (Q1.front()->Freq < Q2.front()->Freq) {
				X = Q1.front();
				Q1.pop();
			} else {
				X = Q2.front();
				Q2.pop();
			}
		}
		if (Q1.size() == 0) {
			Y = Q2.front();
			Q2.pop();
		} else
		if (Q2.size() == 0) {
			Y = Q1.front();
			Q1.pop();
		} else {
			if (Q1.front()->Freq < Q2.front()->Freq) {
				Y = Q1.front();
				Q1.pop();
			} else {
				Y = Q2.front();
				Q2.pop();
			}
		}
		Z = new Node();
		Z->Freq = X->Freq + Y->Freq;
		Z->Left = X;
		Z->Right = Y;
		Q1.push(Z);
	}
}

void DFS(Node *X, int L, long long Code) {
	Node *A = X;
	if (A->Left == 0 && A->Right == 0) {
		Len[A->Id] = L;
		Codes[A->Id] = Code;
		Sol += A->Freq * L;
	} else {
		DFS(A->Left, L+1, (Code<<1));
		DFS(A->Right, L+1, (Code<<1) + 1);
	}
}

int main() {

	freopen("huffman.in","r",stdin);
	freopen("huffman.out","w",stdout);
	scanf("%d", &N);
	
	A.resize(N);
	for (int i = 0; i < N; i++) {
		A[i] = new Node();
		A[i]->Id = i;
		scanf("%d", &A[i]->Freq);
		Q2.push(A[i]);
	}
	
	Huffman();
	DFS(Q1.front(), 0, 0);

	printf("%lld\n", Sol);
	for (int i = 0; i < N; i++) {
		printf("%d %lld\n", Len[i], Codes[i]);
	}
	
	return 0;
}