Cod sursa(job #2580479)

Utilizator Senth30Denis-Florin Cringanu Senth30 Data 13 martie 2020 17:49:26
Problema Loto Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.69 kb
#include <bits/stdc++.h>

using namespace std;
const int NMAX = 110;

struct obj{
    long long val;
    int x, y, z;
}temp[NMAX * NMAX * NMAX], sum[NMAX * NMAX * NMAX];

int N, S, K, M;
int v[NMAX], ans[7];

void read(){
    scanf("%d%d", &N, &S);
    for(int i = 1; i <= N; i++)
        scanf("%d", &v[i]);
}

void gen(int k, int pos){
    if(pos > 3){
        sum[++M] = temp[k];
        return;
    }
    for(int i = 1; i <= N; i++){
        temp[++K] = temp[k];
        temp[K].val += v[i];
        if(pos == 2)
            temp[K].y = i;
        else temp[K].z = i;
        gen(K, pos + 1);
    }
}

bool cmp(obj a, obj b){
    return a.val < b.val;
}

inline int binsearch(int val){
    int st = 1, dr = M;
    int mid, last;
    while(st <= dr){
        mid = (st + dr) / 2;
        if(sum[mid].val <= val){
            last = mid;
            st = mid + 1;
        } else dr = mid - 1;
    }
    if(sum[last].val == val)
        return last;
    else return 0;
}

int main(){

    freopen("loto.in", "r", stdin);
    freopen("loto.out", "w", stdout);

    read();

    for(int i = 1; i <= N; i++){
        temp[++K].val = v[i];
        temp[K].x = i;
        gen(K, 2);
    }
    sort(sum + 1, sum + M + 1, cmp);

    for(int i = 1; i <= M; i++){
        int pos = binsearch(S - sum[i].val);
        if(pos){
            ans[1] = sum[i].x; ans[2] = sum[i].y; ans[3] = sum[i].z;
            ans[4] = sum[pos].x; ans[5] = sum[pos].y; ans[6] = sum[pos].z;
            sort(ans + 1, ans + 7);
            for(int i = 1; i <= 6; i++)
                printf("%d ", ans[i]);
            return 0;
        }
    }
    printf("-1");

    return 0;
}