Editing 3718 - Tort2 (section)

== Rezolvare == 
<syntaxhighlight lang="python" line="1">
# 3718 - Tort2
import sys


def validare_input(n, secventa):
    if n < 2 or n > 200000:
        return False
    for numar in secventa:
        if numar < 1 or numar > 400000:
            return False
    return True


def poate_imparti_secventa(secventa, n, suma_dorita, index, bucata_curenta, prieteni):
    if prieteni == 1:
        return False

    if bucata_curenta == suma_dorita:
        if prieteni == 2:
            return True
        return poate_imparti_secventa(secventa, n, suma_dorita, 0, 0, prieteni - 1)

    for i in range(index, n):
        if secventa[i] + bucata_curenta <= suma_dorita:
            if poate_imparti_secventa(secventa, n, suma_dorita, i + 1, secventa[i] + bucata_curenta, prieteni):
                return True

    return False


def rezolvare(n, secventa):
    suma_totala = sum(secventa)

    for i in range(2, n + 2):
        if suma_totala % i != 0:
            continue

        suma_dorita = suma_totala // i
        if secventa[0] > suma_dorita:
            continue

        if poate_imparti_secventa(secventa, n, suma_dorita, 1, secventa[0], i):
            return i - 1

    return 0


if __name__ == "__main__":
    with open("tort.in", "r") as fin:
        dimensiune = int(fin.readline().strip())
        secventa = list(map(int, fin.readline().split()))

        if not validare_input(dimensiune, secventa):
            print("Datele nu corespund restricțiilor impuse.")
            sys.exit(0)

        print("Datele sunt introduse corect.")
        with open("tort.out", "w") as fout:
            ans = rezolvare(dimensiune, secventa)
            fout.write(str(ans))





</syntaxhighlight>