2904 - Sum Triunghiular

Cerință

Se dă un număr natural n. Să se determine dacă numărul se poate scrie ca sumă de două numere triunghiulare. Dacă este posibil se vor afișa două numere triunghiulare a căror sumă este egală cu n, în orice ordine, respectiv mesajul NU în caz contrar.

Un număr triunghiular este numărul de puncte dintr-un triunghi echilateral umplut uniform cu puncte. De exemplu, 1, 3, 6, 10, 15 sunt numere triunghiulare.

Date de intrare

Programul citește de la tastatură numărul n.

Date de ieșire

Programul va afișa pe ecran, în orice ordine, două numere triunghiulare care însumate dau numărul n, separate printr-un spațiu, sau mesajul NU dacă numărul nu poate fi scris ca suma de două numere triunghiulare.

Restricții și precizări

• n ≤ pow(10,12);
• numerele triunghiulare determinate pot fi identice.

Exemplul 1:

Intrare

24

Ieșire

Datele de intrare corespund restrictiilor impuse.

3 21

Exemplul 2:

Intrare

15

Ieșire

Datele de intrare corespund restrictiilor impuse.

NU

Exemplul 3:

Intrare

dfsgs

Ieșire

Datele de intrare nu corespund restrictiilor impuse.

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>