Cod sursa(job #2735040)

Utilizator HadircaDionisieHadirca Dionisie HadircaDionisie Data 1 aprilie 2021 19:14:39
Problema Loto Scor 25
Compilator py Status done
Runda Arhiva de probleme Marime 0.78 kb
with open('loto.in') as f:
    x = [int(i) for i in f.readline().split()]
    cnt = x[0]
    sum = x[1]

    nums = [int(x) for x in f.readline().split()]

d = {};

n = len(nums)
def Run():
    global nums
    global d
    for i in range(n):
        for j in range(n):
            for z in range(n):
                x = nums[i]+nums[j]+nums[z];
                if x not in d:
                    d[x] = [[nums[i],nums[j],nums[z]]]
                else:
                    d[x].append([nums[i],nums[j],nums[z]])

                if sum - x in d:
                    return[*d[sum-x][0],nums[i],nums[j],nums[z]]


rez = Run()
with open('loto.out','w') as g:
    if rez == -1:
        g.write('-1');
    else:
        for x in rez:
            g.write(str(x)+' ')