Editing 4251 - numereOJI

Gigel este un mare pasionat al cifrelor. Orice moment liber şi-l petrece jucându-se cu numere. Jucându-se astfel, într-o zi a scris pe hârtie <code>10</code> numere distincte de câte două cifre şi a observat că printre acestea există două submulţimi disjuncte de sumă egală. Desigur, Gigel a crezut că este o întâmplare şi a scris alte <code>10</code> numere distincte de câte două cifre şi spre surpriza lui, după un timp a găsit din nou două submulţimi disjuncte de sumă egală.

= Cerința =
Date <code>10</code> numere distincte de câte două cifre, determinaţi numărul de perechi de submulţimi disjuncte de sumă egală care se pot forma cu numere din cele date, precum şi una dintre aceste perechi pentru care suma numerelor din fiecare dintre cele două submulţimi este maximă.

= Date de intrare =
Fișierul de intrare <code>input.txt</code> conține pe prima linie 10 numere naturale distincte separate prin câte un spaţiu <code>x1 x2 ... x10</code>.

= Date de ieșire =
Fișierul de ieșire <code>output.txt</code> va conține trei linii. Pe prima linie se află numărul de perechi de submulţimi de sumă egală, precum şi suma maximă obţinută, separate printr-un spaţiu. Pe linia a doua se află elementele primei submulţimi, iar pe linia a treia se află elementele celei de a doua submulţimi, separate prin câte un spaţiu:

* <code>NrSol Smax</code>, unde <code>NrSol</code> – numărul de perechi; <code>Smax</code> – suma maximă
* <code>x1 ... xk</code> – elementele primei submulţimi
* <code>y1 ... yp</code> - elementele celei de a doua submulţimi

= Restricții și precizări =

* Ordinea submulţimilor în perechi nu contează.
* Perechea de submulţimi determinată nu este obligatoriu unică.

== Exemplul 1 ==
input.txt:

60 49 86 78 23 97 69 71 32 10

output.txt:

65 276

60 49 86 71 10 

78 97 69 32 

Explicație:

Sunt <code>65</code> de soluţii; suma maximă este <code>276</code>, s-au folosit <code>9</code> din cele <code>10</code> numere; prima submulţime are <code>4</code> elemente, a doua are <code>5</code> elemente.

== Rezolvare ==
<syntaxhighlight lang="python3" line="1">
with open("input.txt", "r") as fin, open("output.txt", "w") as fout:
    n = 10
    v = [0] * 12
    a = [0] * 12
    sol = [0] * 12

    i, s1, s2, nrSubm, smax = 0, 0, 0, 0, 0

    l=list(map(int, fin.readline().split()))

    for i in range(1, n + 1):
        v[i] = l[i-1]

    while a[0] == 0:
        # Calculating sums
        s1, s2 = 0, 0
        for i in range(1, n + 1):
            if a[i] == 1:
                s1 += v[i]
            elif a[i] == 2:
                s2 += v[i]

        if s1 == s2:
            nrSubm += 1
            if s1 > smax:
                smax = s1
                sol = a[:]

        # Generating the next sequence of 0, 1, 2
        i = n
        while a[i] == 2:
            a[i] = 0
            i -= 1
        a[i] += 1

    nrSubm //= 2

    fout.write(f"{nrSubm} {smax}\n")

    for i in range(1, n + 1):
        if sol[i] == 1:
            fout.write(f"{v[i]} ")

    fout.write("\n")

    for i in range(1, n + 1):
        if sol[i] == 2:
            fout.write(f"{v[i]} ")

    fout.write("\n")

</syntaxhighlight>