Cod sursa(job #2735036)

Utilizator HadircaDionisieHadirca Dionisie HadircaDionisie Data 1 aprilie 2021 19:08:12
Problema Loto Scor 10
Compilator py Status done
Runda Arhiva de probleme Marime 0.95 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)
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]])


def reRun():
    global nums
    global d
    for i in range(len(nums)-1,-1,-1):
        for j in range(len(nums)-1,-1,-1):
            for z in range(len(nums)-1,-1,-1):
                value = nums[i]+nums[j]+nums[z]
                if sum - value in d:
                    return [*d[sum-value][0],nums[i],nums[j],nums[z]];
    return -1;

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