Cod sursa(job #2932380)

Utilizator acostin643costin andrei acostin643 Data 2 noiembrie 2022 19:24:44
Problema Loto Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.97 kb
#include <fstream>
#include <vector>
#include <iostream>

using namespace std;

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

const int mod = 666013;
vector <int> h[mod + 5];
int b[105];
int sum[1000005];
int p;
int n, s;
bool ok = 0;

void update(int x)
{
    int poz = x % mod;
    h[poz].push_back(x);
}

void discover(int sum1, int sum2)
{
    bool v = 0;
    for(int i = 1; i <= n && !v; i++)
    {
        for(int j = i; j <= n && !v; j++)
        {
            for(int k = j; k <= n && !v; k++)
            {
                if(b[i] + b[j] + b[k] == sum1)
                {
                    fout << b[i] << " " << b[j] << " " << b[k] << " ";
                    v = 1;
                }
            }
        }
    }
    v = 0;
    for(int i = 1; i <= n && !v; i++)
    {
        for(int j = i; j <= n && !v; j++)
        {
            for(int k = j; k <= n && !v; k++)
            {
                if(b[i] + b[j] + b[k] == sum2)
                {
                    fout << b[i] << " " << b[j] << " " << b[k];
                    v = 1;
                }
            }
        }
    }
}

void check(int x)
{
    int k = s - x;
    if(k < 0)
        return;
    int poz = k % mod;
    for(int i = 0; i < h[poz].size(); i++)
    {
        if(h[poz][i] == k)
        {
            discover(x, k);
            ok = 1;
            break;
        }
    }
}

int main()
{
    fin >> n >> s;
    for(int i = 1; i <= n; i++)
    {
        fin >> b[i];
    }

    for(int i = 1; i <= n; i++)
    {
        for(int j = i; j <= n; j++)
        {
            for(int k = j; k <= n; k++)
            {
                int suma = b[i] + b[j] + b[k];
                update(suma);
                sum[p++] = suma;
            }
        }
    }

    for(int i = 0; i < p && !ok; i++)
    {
        check(sum[i]);
    }

    if(!ok)
        fout << "-1";

    fin.close();
    fout.close();
    return 0;
}