Cod sursa(job #3128522)

Utilizator arobyRobert Acsente aroby Data 9 mai 2023 19:41:04
Problema Loto Scor 0
Compilator py Status done
Runda Arhiva de probleme Marime 0.95 kb
from collections import defaultdict

with open("loto.in") as f:
    n, s = map(int, f.readline().split())
    v = list(map(int, f.readline().split()))

m = defaultdict(lambda: {'num1': -1, 'num2': -1, 'num3': -1})

for i in range(n):
    for j in range(n):
        for k in range(n):
            total = v[i] + v[j] + v[k]
            m[total]['num1'] = i
            m[total]['num2'] = j
            m[total]['num3'] = k

with open("loto.out", "w") as f:
    found = False
    for i in range(n):
        for j in range(n):
            for k in range(n):
                total = v[i] + v[j] + v[k]
                if s - total in m:
                    f.write(f"{v[i]} {v[j]} {v[k]} {v[m[s-total]['num1']]} {v[m[s-total]['num2']]} {v[m[s-total]['num3']]}\n")
                    found = True
                    break
            if found:
                break
        if found:
            break

    if not found:
        f.write("-1\n")