Editing 0608 - NkX (section)

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