Условие

Дана квадратная матрица целых чисел mat размера n × n. Требуется вернуть сумму элементов главной и побочной диагоналей. Если n нечётно, центральный элемент считается только один раз.

Решение

class Solution {
    fun diagonalSum(mat: Array<IntArray>): Int {
        var sum = 0
        val n = mat.size
        for (i in 0 until n) {
            sum += mat[i][i] + mat[i][n - 1 - i]
        }
        if (n % 2 == 1) sum -= mat[n/2][n/2]
        return sum
    }
}