Cod sursa(job #1701546)

Utilizator armandpredaPreda Armand armandpreda Data 13 mai 2016 13:43:59
Problema Coduri Huffman Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.3 kb
#include <cstdio>

using namespace std;

const int LIM=2000005;
int n, id=1, p, u, lvl=-1;
long long val, tot;
struct nod
{
    long long val;
    int st, dr;
} arb[LIM];
struct cell
{
    long long val;
    int lvl;
} ans[LIM/2];
void preord(int nod)
{
    lvl++;
    if(nod<=n)
    {
        ans[nod].lvl=lvl;
        ans[nod].val=val;
        tot+=(long long)ans[nod].lvl*arb[nod].val;
        lvl--;
        return;
    }
    val<<=1;
    preord(arb[nod].st);
    val++;
    preord(arb[nod].dr);
    val>>=1;
    lvl--;
}
int main()
{
    freopen("huffman.in", "r", stdin);
    freopen("huffman.out", "w", stdout);
    scanf("%d", &n);
    for(int i=1; i<=n; ++i)
        scanf("%d", &arb[i].val);
    p=n+1, u=n;
    for(int i=1; i<n; ++i)
    {
        int fius, fiud;
        if(p>u or (id<=n and arb[id].val<arb[p].val))
            fius=id, id++;
        else
            fius=p, p++;
        if(p>u or (id<=n and arb[id].val<arb[p].val))
            fiud=id, id++;
        else
            fiud=p, p++;
        u++;
        arb[u].val=arb[fius].val+arb[fiud].val;
        arb[u].st=fius;
        arb[u].dr=fiud;
    }
    preord(2*n-1);
    printf("%lld\n", tot);
    for(int i=1; i<=n; ++i)
        printf("%d %lld\n", ans[i].lvl, ans[i].val);
    return 0;
}