Cod sursa(job #1999088)

Utilizator B_RazvanBaboiu Razvan B_Razvan Data 10 iulie 2017 10:50:49
Problema Coduri Huffman Scor 0
Compilator cpp Status done
Runda Arhiva educationala Marime 1.38 kb
#include <iostream>
#include <cstdio>
#include <queue>
#define NMAX 1000005

using namespace std;

int n, nrTati=1, sol;
pair <int, int> tati[NMAX];
queue <pair <long, int> >q1, q2;


void read()
{
    scanf("%d", &n);
    int x;
    for(int i=1; i<=n; ++i)
    {
        scanf("%d", &x);
        q1.push(make_pair(x, i));
    }
}

pair <long long, long long> minim()
{
    pair <long long, long long> x;
    if(q1.empty())
    {
        x=q2.front();
        q2.pop();
        return x;
    }
    if(q2.empty())
    {
        x=q1.front();
        q1.pop();
        return x;
    }
    if(q1.front() <= q2.front())
    {
        x=q1.front();
        q1.pop();
        return x;
    }
    else
    {
        x=q2.front();
        q2.pop();
        return x;
    }
}

void codare()
{
    pair <long long, long long> x1, x2;
    while(q1.size() + q2.size() > 1)
    {
        x1 = minim();
        x2 = minim();

        q2.push(make_pair(x1.first + x2.first, ++n));
        sol += x1.first + x2.first;

        tati[nrTati].first = n;
        tati[nrTati].second = 0;
        nrTati++;
        tati[nrTati].first = n;
        tati[nrTati].second = 1;
        nrTati++;
    }
}

int main()
{
    freopen("huffman.in", "r", stdin);
    freopen("huffman.out", "w", stdout);
    read();
    codare();
    printf("%d", sol);
    return 0;
}