0855 - Piramida 2
Cerința
Gigel a descoperit planul unei piramide magice. Planul este reprezentat sub forma unei matrice pătratice de dimensiune n, unde n este impar, în care elementele nule nu aparțin piramidei, iar elementele nenule reprezintă înălțimea piramidei în punctul respectiv.
Vezi exemplul pentru detalii!
Pentru n dat, construiți o matrice care să reprezinte planul unei piramide magice.
Date de intrare
Programul citește de la tastatură numărul n.
Date de ieșire
Programul va afișa pe ecran elementele matricei construite, câte o linie a matricei pe o linie a ecranului, elementele de pe o linie fiind separate prin exact un spaţiu
Restricții și precizări
- 3 ⩽ n ⩽ 101, impar
Exemplu 1
- Intrare
- 5
- Ieșire
- 0 0 1 0 0
- 0 1 2 1 0
- 1 2 3 2 1
- 0 1 2 1 0
- 0 0 1 0 0
Exemplu 2
- Intrare
- 2
- Ieșire
- Nu au fost respectate cerintele impuse
Rezolvare
<syntaxhighlight lang="python" line>
- 0855 - Piramida 2
def is_valid_input(n):
return 3 <= n <= 101 and n % 2 == 1
def build_pyramid_matrix(n):
if not is_valid_input(n):
print("Nu au fost respectate cerintele impuse.")
return
pyramid_matrix = [[0] * n for _ in range(n)]
for i in range(n):
for j in range(n):
pyramid_matrix[i][j] = max(abs(n // 2 - i), abs(n // 2 - j))
return pyramid_matrix
def print_matrix(matrix):
for row in matrix:
print(" ".join(map(str, row)))
if __name__ == "__main__":
n = int(input("Introduceți n: "))
pyramid_matrix = build_pyramid_matrix(n)
if pyramid_matrix:
print_matrix(pyramid_matrix)
</syntaxhighlight>