← К задачам
Средне · +3МассивыМатематикаОсновы языка

Отследить эту матрицу.

Дана квадратная матрица (т.е. количество строк и столбцов одинаково), ее след - это сумма элементов на главной диагонали (т.е. диагональная линия от верхнего левого угла до нижнего правого).

Например, для:

[
  [1, 2, 3],
  [4, 5, 6],
  [7, 8, 9]
]

... след равен 1 + 5 + 9 = 15.

Напишите функцию, которая принимает квадратную матрицу и вычисляет ее след.

Примеры

trace([
  [1, 4],
  [4, 1]
]) ➞ 2

# 1 + 1 = 2

trace([
  [1, 2, 3],
  [4, 5, 6],
  [7, 8, 9]
]) ➞ 15

# 1 + 5 + 9 = 15

trace([
  [1, 0, 1, 0],
  [0, 2, 0, 2],
  [3, 0, 3, 0],
  [0, 4, 0, 4]
]) ➞ 10

# 1 + 2 + 3 + 4 = 10

Примечание

Как в примерах, размер матриц будет разным (но они всегда будут квадратными).

def trace(lst):
    
Для запуска тестов необходима авторизация.
Поддержать проект