Сумма по главной диагонали

Матрицы Легкая

Дана квадратная матрица matrix. Верните сумму элементов на главной диагонали.

Сигнатура функции

diagonal_sum(matrix: list[list[int]]) -> int

Примеры

Вход	Ожидаемый результат
[[[1, 2], [3, 4]]]	5
[[[1, 2, 3], [4, 5, 6], [7, 8, 9]]]	15

1решили

1пытались

100%успешность