Латинским квадратом порядка nn называется квадратная матрица размером n×n, каждая строка и каждый столбец которой содержат все числа от 1 до n. Напишите программу, которая проверяет, является ли заданная квадратная матрица латинским квадратом.
Формат входных данных
На вход программе подаётся натуральное число n — количество строк и столбцов в матрице, затем элементы матрицы: n строк, по n чисел в каждой, разделённые пробелами.
Формат выходных данных
Программа должна вывести слово YES, если матрица является латинским квадратом, и слово NO, если не является.
import numpy as np
n = int(input())
matrix = [list(map(int, input().split())) for _ in range(n)]
m = np.array(matrix)
result = 'YES'
for i in range(n):
for count in range(1, n + 1):
if count not in m[:, i]:
result = 'NO'
if count not in m[i]:
result = 'NO'
print(result)
