Cod sursa(job #1020179)

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

const int N = 101, MEM = 101 * 101 * 101;
int num[MEM], n, s, L[MEM], R[MEM];
int SOL[6];
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 = 0; 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) {
        it = ord.find (s - num[i]);
        if (it != ord.end() && num[(*it).second] + num[i] == s) {
            int sol = (*it).second;
            SOL[0] = num[L[i]], SOL[1] = num[L[R[i]]], SOL[2] = num[R[R[i]]];
            SOL[3] = num[L[sol]], SOL[4] = num[L[R[sol]]], SOL[5] = num[R[R[sol]]];
            for (int i = 0; i < 6; ++i)
                fout << SOL[i] << " ";
            return 0;
        }
    }
    fout << -1;
}