Editing 2904 - Sum Triunghiular (section)

==Rezolvare==

<syntaxhighlight lang="python" line="1" start="1">

def numar_triunghiular(numar):    # funcția pentru a calcula al n-lea număr triunghiular
    return numar * (numar + 1) // 2


def doua_numere_triunghiulare(numar):    # funcția pentru a verifica dacă un număr poate fi scris ca suma a două numere triunghiulare
    i = 1
    while numar_triunghiular(i) < numar:
        j = 1
        while numar_triunghiular(j) <= numar - numar_triunghiular(i):
            if numar_triunghiular(i) + numar_triunghiular(j) == numar:
                return numar_triunghiular(i), numar_triunghiular(j)
            j += 1
        i += 1
    return "NU"


def verificare_restrictii(numar):    # funcția de verificare a datelor de intrare
    if 0 < numar <= 10**12:    # adăugăm restricția pentru n
        return True
    else:
        return False


if __name__ == "__main__":
    try:
        n = int(input("Introduceti numarul: "))    # citirea numărului
        if verificare_restrictii(n):    # verificăm datele de intrare
            print("Datele de intrare corespund restrictiilor impuse.")
            rezultat = doua_numere_triunghiulare(n)    # apelăm funcția pentru a verifica dacă numărul poate fi scris ca suma a două numere triunghiulare
            if rezultat == "NU":
                print(rezultat)
            else:
                print(*rezultat)
    except ValueError:    # tratăm cazul în care utilizatorul introduce un input invalid
        print("Datele de intrare nu corespund restrictiilor impuse.")

</syntaxhighlight>