Editing 4245 - Suma ONI (section)

== Rezolvare == 
<syntaxhighlight lang="python" line>
#4245
def verifica_date_intrare(s):
    if s < 1 or s > 100000:
        print("Datele introduse nu corespund cerintelor.")
        exit()


def cauta_suma_minimala(s):
    start_interval = 1
    final_interval = 1000
    suma_minima_gasita = 0

    while start_interval <= final_interval:
        mijloc_interval = start_interval + (final_interval - start_interval) // 2
        suma = mijloc_interval * (mijloc_interval + 1) // 2
        if suma >= s:
            suma_minima_gasita = mijloc_interval
            final_interval = mijloc_interval - 1
        else:
            start_interval = mijloc_interval + 1

    diferenta = suma_minima_gasita * (suma_minima_gasita + 1) // 2 - s
    while diferenta & 1:
        suma_minima_gasita += 1
        diferenta = suma_minima_gasita * (suma_minima_gasita + 1) // 2 - s

    return suma_minima_gasita, diferenta


def afiseaza_rezultat(suma_minima_gasita, diferenta):
    fisier_iesire = open("sumaoni.out", "w")
    fisier_iesire.write(str(suma_minima_gasita) + '\n')
    if diferenta == 0:
        fisier_iesire.close()
        return

    if diferenta // 2 <= suma_minima_gasita:
        fisier_iesire.write(str(diferenta // 2))
        fisier_iesire.close()
        return

    diferenta //= 2
    for i in range(1, diferenta // 2 + 1):
        if diferenta - i <= suma_minima_gasita and diferenta - i != i:
            fisier_iesire.write(str(i) + '\n' + str(diferenta - i))
            fisier_iesire.close()
            return

    fisier_iesire.close()


if __name__ == '__main__':
    fisier_intrare = open("sumaoni.in", "r")
    s = int(fisier_intrare.readline())
    fisier_intrare.close()

    verifica_date_intrare(s)

    suma_minima_gasita, diferenta = cauta_suma_minimala(s)

    if suma_minima_gasita is not None and diferenta is not None:
        print("Datele introduse corespund cerintelor.")
        afiseaza_rezultat(suma_minima_gasita, diferenta)

</syntaxhighlight>