Задача к ЕГЭ по информатике на тему «Подготовка к 27» №2

Просмотров 10 Комментарии 0

На вход программе подаётся натуральное число $N$ — количество строк и столбцов в двумерном массиве, затем элементы двумерного массива (целые числа) построчно через пробел. Найдите сумму всех элементов, которые имеют одинаковый первый и второй индекс (или иными словами сумму элементов главной диагонали матрицы).

f = open(’1.txt’)
n = int(f.readline())
a = []

for i in range(n):
    a.append([int(x) for x in f.readline().split()])

s = 0
for i in range(n):
    s += a[i][i]

print(s)

Ответ: 556