0608 - NkX

Cerinţa

Se dau două numere naturale n și k, cu proprietatea că 2*k < n. Să se afişeze pe ecran elementele unei matrice pătratice de dimensiune n, formată din valorile 1 şi 2, construită astfel:

valorile de pe diagonala principală sunt 1;
valorile de pe k pseudodiagonale paralele cu diagonala principală şi situate imediat deasupra acesteia sunt 1;
valorile de pe k pseudodiagonale paralele cu diagonala principală şi situate imediat sub aceasta sunt 1;
valorile de pe diagonala secundară sunt 1;
valorile de pe k pseudodiagonale paralele cu diagonala secundară şi situate imediat deasupra acesteia sunt 1;
valorile de pe k pseudodiagonale paralele cu diagonala secundară şi situate imediat sub aceasta sunt 1;
restul valorilor din matrice sunt 2.

Date de intrare

Programul citește de la tastatură numerele n k.

Date de ieșire

Programul va afișa pe ecran matricea construită, câte o linie a matrice pe o linie a ecranului, elementele de pe o line fiind separate prin exact un spaţiu.

Restricţii şi precizări

1 ⩽ n ⩽ 100

Exemplu

Intrare: 10 1
Ieșire: Datele de intrare corespund restrictiilor impuse; 1 1 2 2 2 2 2 2 1 1; 1 1 1 2 2 2 2 1 1 1; 2 1 1 1 2 2 1 1 1 2; 2 2 1 1 1 1 1 1 2 2; 2 2 2 1 1 1 1 2 2 2; 2 2 2 1 1 1 1 2 2 2; 2 2 1 1 1 1 1 1 2 2; 2 1 1 1 2 2 1 1 1 2; 1 1 1 2 2 2 2 1 1 1; 1 1 2 2 2 2 2 2 1 1

Exemplu 2

Intrare: 121 210
Ieșire: Datele de intrare nu corespund restrictiilor impuse

Rezolvare

<syntaxhighlight lang="python" line> def validare(nr, bare1): # functia de validare a datelor de intrare

   if not isinstance(nr, int) or nr < 1 or nr > 100:
       raise ValueError("Datele de intrare nu corespund restrictiilor impuse")

   if not isinstance(bare1, int) or bare1 < 0 or bare1 > 50:
       raise ValueError("Datele de intrare nu corespund restrictiilor impuse")

   print("Datele de intrare corespund restrictiilor impuse")

def main():

   # Solicităm introducerea numerelor n și k
   print("Introduceți numerele n și k, separate printr-un spațiu: ")
   n, k = map(int, input().split())

   try:
       validare(n, k)  # apelul functiei de validare
   except ValueError as e:
       print(e)
       return

   # Inițializăm matricea cu valori de 2
   matrice = [[2] * n for _ in range(n)]

   # Parcurgem fiecare element din matrice
   for i in range(n):
       for j in range(n):
           # Dacă elementul se află pe una dintre diagonalele specificate, îl setăm la 1
           if abs(i - j) <= k or abs(i + j - n + 1) <= k:
               matrice[i][j] = 1

   # Afișăm matricea, fiecare linie pe o linie nouă și elementele separate prin spațiu
   for linie in matrice:
       print(' '.join(map(str, linie)))

if __name__ == "__main__":

   main()

</syntaxhighlight>