Cod sursa(job #1606881)

Utilizator andrew_assassin789Andrei Manea andrew_assassin789 Data 20 februarie 2016 17:25:27
Problema Coduri Huffman Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 1.83 kb
#include <fstream>
#include <queue>
#include <climits>
#define uni unsigned long long
#define w 1000001
using namespace std;
queue <uni> q1;
queue <uni> q2;
uni frec[w];
uni v[w][2];
uni sol;uni lg[w],b[w];
void df(uni x, uni cod, uni niv)
{
    if (v[x][0])
    {
        df(v[x][0],2*cod,niv+1);
        df(v[x][1],2*cod+1,niv+1);
        return;
    }
    lg[x]=niv;
    b[x]=cod;
}
void huff(uni n)
{
    uni i,j,k=n,min,p,x;
    for (j=1;j<=n-1;j++)
    {
        k++;
        /* adaugam un nod k, tatal celor mai
        mici 2 elemente din cele 2 cozi*/
        for (i=0;i<=1;i++)
        {
            min=LLONG_MAX;
            if (!q1.empty())
            {
                x=q1.front();
                if (frec[x]<min)
                {
                    min=frec[x];
                    p=1;
                }
            }
            if (!q2.empty())
            {
                x=q2.front();
                if (frec[x]<min)
                {
                    min=frec[x];
                    p=2;
                }
            }
            if (p==1)
            {
                x=q1.front();
                q1.pop();
                v[k][i]=x;
                frec[k]+=frec[x];
            }
            else
            {
                x=q2.front();
                q2.pop();
                v[k][i]=x;
                frec[k]+=frec[x];
            }
        }
        sol+=frec[k];
        q2.push(k);
    }
    df(k,0,0);
}
int main()
{
    ifstream f("huffman.in");
    ofstream g("huffman.out");
    uni i,n;
    f>>n;
    for (i=1;i<=n;i++)
    {
        f>>frec[i];
        q1.push(i);
    }
    huff(n);
    g<<sol<<'\n';
    for (i=1;i<=n;i++)
    {
        g<<lg[i]<<' '<<b[i]<<'\n';
    }
    f.close();
    g.close();
    return 0;
}