Cod sursa(job #1020186)

Utilizator tudorv96Tudor Varan tudorv96 Data 1 noiembrie 2013 19:37:57
Problema Loto Scor 75
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.57 kb
#include <fstream>
#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
#include <cassert>
using namespace std;

const int N = 101, MEM = N * N * N;
int num[MEM], n, s, L[MEM], R[MEM];
map <int, int> ord;

ifstream fin ("loto.in");
ofstream fout ("loto.out");

int main() {
    fin >> n >> s;
    int MAX = -1, MIN = 2e9;
    for (int i = 0; i < n; ++i) {
        fin >> num[i];
        MIN = min (MIN, num[i]);
        MAX = max (MAX, num[i]);
    }
    if (MAX * 6 < s || MIN * 6 > s) {
        fout << -1;
        return 0;
    }
    fin.close();
    for (int i = 0; i < n; ++i)
        for (int j = i; j < n; ++j) {
            int nod = n * (i + 1) + j;
            num[nod] = num[i] + num[j];
            L[nod] = i;
            R[nod] = j;
        }
    for (int i = 0; i < n; ++i)
        for (int j = n; j < n * (n + 1); ++j) {
            int nod = n * (n + 1) + i * n * n + (j - n);
            num[nod] = num[i] + num[j];
            L[nod] = i;
            R[nod] = j;
            ord.insert (make_pair (num[nod], nod));
        }
    map <int, int> :: iterator it;
    for (int i = n * (n + 1); i < n * n * n + n * n + n; ++i) {
        if (num[i] > s)
            continue;
        it = ord.find (s - num[i]);
        if (it != ord.end() && num[(*it).second] + num[i] == s) {
            int sol = (*it).second;
            fout << num[L[i]] << " " << num[L[R[i]]] << " " << num[R[R[i]]] << " " << num[L[sol]] << " " << num[L[R[sol]]] << " " << num[R[R[sol]]];
            return 0;
        }
    }
    fout << -1;
}