👩💻 Решение задачи на Python про сумму диагоналей матрицы
Задача: Дана матрица mat. Напишите функцию, которая возвращает сумму элементов диагоналей матрицы.
Первым делом следует уточнить у интервьюера, какие есть ограничения. Например, он сказал, что на вход мы получаем только квадратную матрицу, элементы на пересечении диагоналей считаются только один раз, элементы могут принимать значения от 1 до 100.
Тогда решение может выглядеть так:
🧩 Мы будем проходиться в цикле и сразу прибавлять к результату значения матрицы по индексам i (
Это будет решение за O(n), так как мы один раз проходимся в цикле по элементам матрицы.
#junior
#middle
Задача: Дана матрица mat. Напишите функцию, которая возвращает сумму элементов диагоналей матрицы.
Первым делом следует уточнить у интервьюера, какие есть ограничения. Например, он сказал, что на вход мы получаем только квадратную матрицу, элементы на пересечении диагоналей считаются только один раз, элементы могут принимать значения от 1 до 100.
Тогда решение может выглядеть так:
🧩 Мы будем проходиться в цикле и сразу прибавлять к результату значения матрицы по индексам i (
mat[i][i]). Дальше будем смотреть: если i не равно n-i-1, где n — размер матрицы, значит мы прибавляем к результату mat[i][n-i-1]. Эта проверка позволит нам избежать повторного прибавления к сумме элементов, которые стоят на пересечении диагоналей. Это будет решение за O(n), так как мы один раз проходимся в цикле по элементам матрицы.
#junior
#middle