Daily LeetCode Challenge - 1572. Matrix Diagonal Sum

Problem From.
https://leetcode.com/problems/matrix-diagonal-sum/

오늘 문제는 이차원 배열 matrix 가 주어졌을때, 각 대각선의 합을 구하는 문제였다.

가운데에 겹치는 원소가 있다면 그 원소를 빼고 합을 구하면 되는 문제로 O(N) 의 시간 복잡도 안에 풀이를 구할 수 있었다.

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