Матрицы. Латинский квадрат

Латинским квадратом порядка 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)




Добавить комментарий

;-) :| :x :twisted: :smile: :shock: :sad: :roll: :razz: :oops: :o :mrgreen: :lol: :idea: :grin: :evil: :cry: :cool: :arrow: :???: :?: :!: