Cod sursa(job #1042580)

Utilizator MoneaVladMonea Vlad MoneaVlad Data 27 noiembrie 2013 13:16:14
Problema Loto Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.32 kb
#include <fstream>
#include <algorithm>

using namespace std;

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

int a[101], I[1000001], J[1000001], K[1000001];
char S[300000007];
int l;

int cautbin (int x){
    int st = 1;
    int dr = l;
    while (dr - 1 > st){
        int mid = st + (dr - st) / 2;
        if (S[mid] == x)
            return mid;
        if (S[mid] < x)
            st = mid + 1;
        else
            dr = mid - 1;
    }
    if (S[dr] == x)
        return dr;
    if (S[st] == x)
        return st;
    else
        return -1;
}

int main()
{
    int N, Suma, i, j, k, poz, ok;
    in >> N >> Suma;
    for (i = 1; i <= N; i++)
        in >> a[i];
    for (i = 1; i <= N; i++)
        for (j = i; j <= N; j++)
            for (k = j; k <= N; k++)
            {
                l++;
                S[l] = a[i] + a[j] + a[k];
                I[l] = i;
                J[l] = j;
                K[l] = k;
            }
    sort(S+1, S+l+1);
    ok = 0;
    for (i = 1; i <= l; i++){
        poz = cautbin (Suma - S[i]);
        if (poz > 0) ok = 1;
        if (poz > 0){
            out << I[i] << " " << J[i] << " " << K[i] << " " << I[poz] << " " << J[poz] << " " << K[poz];
            break;
        }
    }
    if(ok == 0)
        out << -1;
    return 0;
}