Cod sursa(job #1093575)

Utilizator MoneaVladMonea Vlad MoneaVlad Data 28 ianuarie 2014 11:49:57
Problema Loto Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.28 kb
#include <fstream>
#include <iostream>

using namespace std;

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

int a[101], I[1000001], J[1000001], K[1000001];
long long S[10000001];
int l;

int cautbin (int x, int st, int dr){
    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;
            }
    ok = 0;
    for (i = 1; i <= l; i++){
        poz = cautbin (Suma - S[i], 1, l);
        if (poz > 0){
            ok = 1;
            out << I[i] << " " << J[i] << " " << K[i] << " " << I[poz] << " " << J[poz] << " " << K[poz];
            break;
        }
    }
    if(ok == 0)
        out << -1;
    return 0;
}