Cod sursa(job #2624799)

Utilizator ApostolIlieDanielApostol Daniel ApostolIlieDaniel Data 5 iunie 2020 13:43:07
Problema Coduri Huffman Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.23 kb
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
#define pb push_back
#define dbg(x) cerr << #x << " " << x << "\n"

const int N = 1e6;
ll ans[1 + N];
int length[1 + N];
ll freq[1 + N];
int r[1 + N], l[1 + N];

int go (int node, int val, int h) {
    if (l[node] == 0 && r[node] == 0) {
        ans[node] = val;
        length[node] = h;
        return 0;
    }
    return freq[node] + go (l[node], val * 2, h + 1) + go (r[node], val * 2 + 1, h + 1);
}

int main() {
    freopen ("huffman.in", "r", stdin);
    freopen ("huffman.out", "w", stdout);

    ios::sync_with_stdio (false);
    cin.tie (0); cout.tie (0);

    int n;
    cin >> n;
    priority_queue <pair <int, int>> pq;
    for (int i = 1; i <= n; i++) {
        cin >> freq[i];
        pq.push ({-freq[i], i});
    }

    int nr = n;
    while (pq.size () > 1) {
        int x = pq.top ().second;
        pq.pop ();
        int y = pq.top ().second;
        pq.pop ();
        freq[++nr] = freq[x] + freq[y];
        l[nr] = x, r[nr] = y;
        pq.push ({-freq[nr], nr});
    }

    cout << go (nr, 0, 0) << "\n";
    for (int i = 1; i <= n; i++)
        cout << ans[i] << " " << length[i] << "\n";

    return 0;
}