Cod sursa(job #2251471)

Utilizator paul_danutDandelion paul_danut Data 1 octombrie 2018 17:27:07
Problema Coduri Huffman Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.44 kb

#include <fstream>

using namespace std;

#define maxn 1000100
#define INF 1LL * 1000000000 * 1000000000

struct nod
{
    long long val;
    long child[2];
} nod[2*maxn];

long n, i, j, k, p, l[2], r[2];
long q[2][maxn], length[maxn];
long long b[maxn], m, sol;

void df(long poz, long long cod, long nivel)
{
    if(nod[poz].child[0])
    {
        df(nod[poz].child[0], cod*2, nivel+1);
        df(nod[poz].child[1], cod*2+1, nivel+1);
        return;
    }
    length[poz]=nivel;
    b[poz]=cod;
}

void solve()
{
    k=n;
    l[0]=l[1]=1;
    r[0]=n;
    while(l[0]+l[1]<=r[0]+r[1])
    {
        ++k;
        for(j=0; j<2; j++)
        {
            p=0;
            m=INF;
            for(i=0; i<2; i++)
            {
                if(nod[q[i][l[i]]].val<m && l[i]<=r[i])
                {
                    p=i;
                    m=nod[q[i][l[i]]].val;
                }
            }
            nod[k].child[j]=q[p][l[p]];
            nod[k].val+=nod[q[p][l[p]]].val;
            l[p]++;
        }
        sol += nod[k].val;
        q[1][++r[1]]=k;
    }

    df(k, 0, 0);
}

int main()
{
    ifstream f("huffman.in");
    ofstream g("huffman.out");
    f>>n;

    for(i=1; i<=n; i++)
    {
        f>>nod[i].val;
        q[0][i]=i;
    }

    solve();
    g << sol << '\n';

    for(i=1; i<=n; i++)
    {
        g << length[i] << ' ' << b[i] << '\n';
    }

    return 0;
}