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

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=" ")