Cod sursa(job #1427796)

Utilizator retrogradLucian Bicsi retrograd Data 3 mai 2015 01:53:31
Problema Coduri Huffman Scor 80
Compilator cpp Status done
Runda Arhiva educationala Marime 2.24 kb
#include<fstream>
#include<queue>

using namespace std;

ifstream cin("huffman.in");

typedef int64_t var;

#define MAXN 2000005
int PARENT[MAXN];
var V[MAXN];
int BIT[MAXN];
int D[MAXN];
bool COMP[MAXN];


struct Node {
    int n;
    var c;
    Node(int a, var b) {
        n = a;
        c = b;
    }
    Node(){}
    const bool operator<(const Node &n) const {
        return c < n.c;
    }
};


struct _Queue {
    Node Q[3100000];
    var b = 1, e = 1;

    void pop() {b++;}
    bool empty() {return b>=e;}
    Node front() {return Q[b];}
    void push(const Node &val) {Q[e++] = val;}
    int size() {return e-b;}
};
_Queue Q1, Q2;

Node extract_min() {
    if(Q1.empty()) {
        Node rez = Q2.front();
        Q2.pop();
        return rez;
    }
    if(Q2.empty()) {
        Node rez = Q1.front();
        Q1.pop();
        return rez;
    }
    if(Q2.front() < Q1.front()) {
        Node rez = Q2.front();
        Q2.pop();
        return rez;
    } else {
        Node rez = Q1.front();
        Q1.pop();
        return rez;
    }
}

#define DIM 100000
char buff[DIM];
var poz;
void Read(var &a) {
    while(!isdigit(buff[poz]))
        if(++poz == DIM)
            cin.read(buff, DIM), poz=0;
    a = 0;
    while(isdigit(buff[poz])) {
        a = a * 10 + buff[poz] - '0';
        if(++poz == DIM)
            cin.read(buff, DIM), poz=0;
    }
}



int main() {

    var n, v;

    cin.read(buff, DIM);
    freopen("huffman.out", "w", stdout);

    Read(n);
    var _n = n;
    for(int i=1; i<=n; i++) {
        Read(V[i]);
        Q1.push(Node(i, V[i]));
    }

    while(Q1.size() + Q2.size() > 1) {
        auto node1 = extract_min();
        auto node2 = extract_min();

        PARENT[node1.n] = PARENT[node2.n] = ++n;
        BIT[node1.n] = 0;
        BIT[node2.n] = 1;
        Q2.push(Node(n, node1.c + node2.c));
    }

    var sum = 0;
    //BIT[n] = 0;
    for(var i=n-1; i>=1; i--) {
        BIT[i] |= (BIT[PARENT[i]] << 1);
        D[i] = D[PARENT[i]] + 1;
        sum += D[i]*V[i];
    }

    COMP[n] = 1;
    n = _n;

    printf("%lld\n", sum);


    for(var i=1; i<=n; i++) {
        printf("%d %d\n", D[i], BIT[i]);
    }

    return 0;
}