Cod sursa(job #1670472)

Utilizator ZimmyZimmermann Erich Zimmy Data 31 martie 2016 19:18:42
Problema Coduri Huffman Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 1.24 kb
#include <fstream>
#define upi {k++;v[k]=v[i]+v[i+1];s[0][k]=i;s[1][k]=i+1;i+=2;sol1+=v[k];}
#define upj {k++;v[k]=v[j]+v[j+1];s[0][k]=j;s[1][k]=j+1;j+=2;sol1+=v[k];}
#define upd {k++;v[k]=v[i]+v[j];s[0][k]=i;s[1][k]=j;i++;j++;sol1+=v[k];}
#define N 2000010
using namespace std;
ifstream f("huffman.in");
ofstream g("huffman.out");

int s[2][N],n,i,j,k,sol1;
long long v[N],cod[N],doila[30];
int main()
{
    f>>n;
    for(i=1;i<=n;i++)
        f>>v[i];
    k=n;i=1;upi;j=n+1;
    for(;i<n;)
    {
        if(j<k)
        {
            if(v[i+1]<=v[j]&&v[i]+v[i+1]<=v[j]+v[j+1])upi else
            if(v[j+1]<=v[i]&&v[j]+v[j+1]<=v[i]+v[i+1])upj else
            upd
        }
        else
        if(j==k)
        {
             if(v[i+1]<=v[j])upi else
             upd
        }
    }
    for(;i==n;)
        if(j<k)
        {
            if(v[j+1]<=v[i])upj else
            upd
        }
        else
            upd;
    for(;j<k;)
        upj;
    g<<sol1<<'\n';
    v[2*n-1]=0;
    for(i=2*n-1;i>n;i--)
    {
        v[s[0][i]]=v[i]+1;
        cod[s[0][i]]=2*cod[i];
        v[s[1][i]]=v[i]+1;
        cod[s[1][i]]=2*cod[i]+1;
    }
    for(i=1;i<=n;i++)
        g<<v[i]<<' '<<cod[i]<<'\n';
    return 0;
}