Cod sursa(job #791985)

Utilizator repp4raduRadu-Andrei Szasz repp4radu Data 26 septembrie 2012 00:09:32
Problema Coduri Huffman Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.43 kb
#include <fstream>

#define MAX 1048576
#define INF ((1LL << 62) - 1)

using namespace std;

struct nod
{
    long long v;
    long long f[2];
}nod[2 * MAX];

long long sol, b[MAX], minim;
int l[2], r[2], q[2][MAX], n, k, p, lgt[MAX];

void dfs(int nd, long long cod, long long l)
{
    if(nod[nd].f[0])
    {
        dfs(nod[nd].f[0], cod * 2, l + 1);
        dfs(nod[nd].f[1], cod * 2 + 1, l + 1);
        return;
    }
    b[nd] = cod;
    lgt[nd] = l;
}

void solve()
{
    k = n;
    l[0] = l[1] = 1;
    r[0] = n;
    while(l[0] + l[1] <= r[0] + r[1])
    {
        k++;
        for(int i = 0; i < 2; i++)
        {
            p = 0; minim = INF;
            for(int j = 0; j < 2; j++)
            {
                if(l[j] <= r[j] && nod[q[j][l[j]]].v < minim)
                {
                    minim = nod[q[j][l[j]]].v;
                    p = j;
                }
            }
            nod[k].f[i] = q[p][l[p]];
            nod[k].v += nod[q[p][l[p]++]].v;
        }
        sol += nod[k].v;
        q[1][++r[1]] = k;
    }
    dfs(k, 0, 0);
}

int main()
{
    ifstream in("huffman.in");
    in>>n;
    for(int i = 1; i <= n; i++)
    {
        in>>nod[i].v;
        q[0][i] = i;
    }
    in.close();
    solve();
    ofstream out("huffman.out");
    out<<sol<<"\n";
    for(int i = 1; i <= n; i++)
        out<<lgt[i]<<" "<<b[i]<<"\n";
    out.close();
    return 0;
}