1684 - Numere Piramidale

Cerința

Amicul nostru, Zoli, a învățat la scoală despre pătrate perfecte și numere piramidale. Al n-lea număr piramidal înseamnă suma primelor n pătrate perfecte, începând de la 1. Ajutați-l pe Zoli sa afle primele n numere piramidale.

Date de intrare

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

Date de ieșire

Programul va afișa pe ecran primele n numere piramidale, în ordine crescătoare, separate prin câte un spațiu.

Restricții și precizări

1 ≤ n ≤ 400

Exemplu:

Intrare 3

Ieșire 1 5 14

Explicație

Primele 3 numere piramidale sunt 1, 5</span si 14</span.

Rezolvare

<syntaxhighlight lang=“python” line> import math

def is_square(n):

   return int(math.sqrt(n)) ** 2 == n

n = int(input()) suma = 0

for i in range(1, n + 1):

   suma += i ** 2
   print(suma, end=" ")

</syntaxhighlight>