1684 - Numere Piramidale
De la Universitas MediaWiki
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 validate_input(n):
if not n.isdigit():
return False
n = int(n)
if n < 1 or n > 400:
return False
return True
def compute_pyramid_numbers(n):
pyramid_numbers = []
for i in range(1, n+ 1):
pyramid_number = sum([j ** 2 for j in range(1, i + 1)])
pyramid_numbers.append(pyramid_number)
return pyramid_numbers
def main():
n = input("Introduceti numarul de numere piramidale: ")
while not validate_input(n):
n = input("Input invalid! Introduceti numarul de numere piramidale: ")
pyramid_numbers = compute_pyramid_numbers(int(n))
print("Primele", n, "numere piramidale sunt:", " ".join(str(p) for p in pyramid_numbers))
if __name__ == '__main__':
main()