Cod sursa(job #2757988)

Utilizator Andreea__Zavoiu Andreea Andreea__ Data 7 iunie 2021 21:22:27
Problema Loto Scor 95
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.4 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <unordered_map>

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

int main()
{
    int n, sum, val, s, ok=0;
    vector<int> nr;
    unordered_map< int, vector<int> > umap; // arbore care nu e nevoie sa nu fie ordonat

    fin >> n >> sum;
    for (int i = 0; i < n; ++i) {   // citesc numerele de la loto
        fin >> val;
        nr.push_back(val);  // le tin intr-un vector
    }

    // iau fiecare element din vector de la 1 la n, le fac suma pe tupluri de cate 3 si o adaug in map, iar pentru restul de 3 elemente voi cauta in arbore diferenta pana la suma ceruta
    for (int i=0; i < n; ++i)
    {
        for (int j=i; j < n; ++j)
        {
            for (int k=j; k < n; ++k)
            {
                s = nr[i] + nr[j] + nr[k];
                umap[s] = { nr[i], nr[j], nr[k] }; // tuplu cu jumatate din cele 6 numere dorite

                if ( umap.find(sum-s) != umap.end() )   // daca am gasit tuplul care sa-mi dea restul de suma pana la cea ceruta, atunci afisez cele 6 numere
                {
                    fout << nr[i] << " " << nr[j] << " " << nr[k] << " " << umap[sum-s][0] << " " << umap[sum-s][1] << " " << umap[sum-s][2];
                    //ok=1;
                    return 0;
                }
            }
        }
    }
    fout << -1;
    return 0;
}