Cod sursa(job #1670421)

Utilizator ZimmyZimmermann Erich Zimmy Data 31 martie 2016 18:35:39
Problema Coduri Huffman Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.45 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(;;)
    {
        if(i<n&&j<k)
        {
            if(v[i]+v[i+1]<v[i]+v[j]&&v[i]+v[i+1]<v[j]+v[j+1])upi else
            if(v[j]+v[j+1]<v[i]+v[j]&&v[j]+v[j+1]<v[i]+v[i+1])upj else
            upd
            continue;
        }
        if(i<n&&j==k)
        {
            if(v[i]+v[i+1]<v[i]+v[j])upi else
            upd
            continue;
        }
        if(j<k&&i==n)
        {
            if(v[j]+v[j+1]<v[i]+v[j])upj else
            upd
            continue;
        }
        if(j<k){upj continue;}
        if(i<n){upi continue;}
        if(i==n&&j==k){upd continue;}
        break;


    }
    g<<sol1<<'\n';
    doila[0]=1;
    for(i=1;i<=18;i++)doila[i]=2*doila[i-1];
    v[2*n-1]=0;
    for(i=2*n-1;i>n;i--)
    {
        v[s[0][i]]=v[i]+1;
        cod[s[0][i]]=cod[i];
        v[s[1][i]]=v[i]+1;
        cod[s[1][i]]=cod[i]+doila[v[i]];
    }
    for(i=1;i<=n;i++)
        g<<v[i]<<' '<<cod[i]<<'\n';

    return 0;
}