Editing 2106 - Armonica (section)

== Rezolvare ==
<syntaxhighlight lang="python" line="1">
# 2106 - Armonica
def validare(valoaree_b):           # functia de validare a datelor de intrare
    if valoaree_b < 1 or valoaree_b > 1000000000:
        raise ValueError
    fisier_iesire.write("Datele introduse corespund restrictiilor impuse.\n")


def progresie_armonica(valoaree_b):                     # functia de rezolvare
    rezultat = []
    for valoaree_a in range(1, valoaree_b + 1):
        valoaree_c = (valoaree_a * valoaree_b) / (2 * valoaree_a - valoaree_b)
        if valoaree_c.is_integer() and valoaree_c >= valoaree_a:
            rezultat.append((valoaree_a, int(valoaree_c)))
            if valoaree_a != valoaree_c:  # Daca a si c sunt diferite, adaugam si perechea (c, a)
                rezultat.append((int(valoaree_c), valoaree_a))
    return rezultat


if __name__ == '__main__':
    fisier_intrare = open("armonicain.txt", "r")         # declararea fisierelor
    fisier_iesire = open("armonicaout.txt", "w")       # fisierul out trebuie declarat cu optiunea "w" (write)

    # din cauza datelor de intrare pot aparea 2 tipuri de erori, valueError sau IndexError pe care le tratam
    try:
        valoare_b = int(fisier_intrare.readline())      # citirea datelor de intrare

        validare(valoare_b)                 # apelul functiei de validare
        rezultate = progresie_armonica(valoare_b)  # apelul functiei de rezolvare
        rezultate.sort(key=lambda x: x[0] + x[1])  # Sortam solutiile in functie de suma a + c
        fisier_iesire.write(f"{len(rezultate)}\n")
        for valoare_a, valoare_c in rezultate:
            fisier_iesire.write(f"{valoare_a} {valoare_c}\n")

    except ValueError:
        fisier_iesire.write("Datele introduse nu corespund restrictiilor impuse.")

</syntaxhighlight>