Cod sursa(job #1020149)

Utilizator tudorv96Tudor Varan tudorv96 Data 1 noiembrie 2013 18:55:51
Problema Loto Scor 5
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.02 kb
#include <fstream>
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

const int N = 101, MEM = 101 * 101 * 101;
int num[MEM], n, s, ord[MEM], L[MEM], R[MEM];
int SOL[6];

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

bool cmp (int a, int b) {
    return num[a] < num[b];
}

int bs (int x) {
    int lo = 0, hi = n * n * n;
    while (lo <= hi) {
        int mid = (lo + hi) >> 1;
        if (num[ord[mid]] == x)
            return ord[mid];
        else
            if (num[ord[mid]] < x)
                lo = mid + 1;
        else
            hi = mid;
    }
    return -1;
}

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[i * n * n + j - n] = nod;
        }
    sort (ord, ord + n * n * n, cmp);
    for (int i = 0; i < n; ++i)
        for (int j = 0; j < n; ++j)
            for (int k = 0; k < n; ++k) {
                int sol = bs (s - num[i] - num[j] - num[k]);
                if (sol != -1) {
                    SOL[0] = num[i], SOL[1] = num[j], SOL[2] = num[k];
                    SOL[3] = num[L[sol]], SOL[4] = num[L[R[sol]]], SOL[5] = num[R[R[sol]]];
                    sort (SOL, SOL + 6);
                    for (int i = 0; i < 6; ++i)
                        fout << SOL[i] << " ";
                    return 0;
                }
            }
}