Дан массив размера N×N. Необходимо вывести положительные элементы главной диагонали.
Формат входных данных
Натуральное число N — размер массива (N≤50). В следующих N строках вводится сам массив, состоящий из целых чисел, по модулю не превышающих 100.
Формат выходных данных
Требуется вывести положительные элементы главной диагонали или «No such elements», если положительных элементов нет.
Примеры
входные данные выходные данные
4
22 7 -9 23
17 -19 -24 25
20 47 31 -16
-42 -45 29 49
22 31 49
3
-3 18 -36
-16 -1 0
-16 9 -3
No such elements

Предыдущий
вопрос Следующий
вопрос

# Считываем размер массива
N = int(input())
# Инициализируем массив
matrix = []
positive_diagonal_elements = []
# Считываем элементы массива
for _ in range(N):
row = list(map(int, input().split()))
matrix.append(row)
# Находим положительные элементы на главной диагонали
for i in range(N):
if matrix[i][i] > 0:
positive_diagonal_elements.append(matrix[i][i])
# Проверяем наличие положительных элементов на главной диагонали
if not positive_diagonal_elements:
print("No such elements")
else:
for element in positive_diagonal_elements:
print(element, end=' ')