[leetCode/Javascript] 1572. Matrix Diagonal Sum

문제

Given a square matrix mat, return the sum of the matrix diagonals.

Only include the sum of all the elements on the primary diagonal and all the elements on the secondary diagonal that are not part of the primary diagonal.

 

Example 1:

Input: mat = [[1,2,3],
              [4,5,6],
              [7,8,9]]
Output: 25
Explanation: Diagonals sum: 1 + 5 + 9 + 3 + 7 = 25
Notice that element mat[1][1] = 5 is counted only once.

Example 2:

Input: mat = [[1,1,1,1],
              [1,1,1,1],
              [1,1,1,1],
              [1,1,1,1]]
Output: 8

Example 3:

Input: mat = [[5]]
Output: 5

 

Constraints:

• n == mat.length == mat[i].length
• 1 <= n <= 100
• 1 <= mat[i][j] <= 100

 

 

접근방향

row랑 column으로 나뉘어서 두가지의 자리수를 체크해서 해당 자리에 있는 수를 뽑아서 배열에 넣고, 그 배열의 합산을 구하면 되겠다라고 생각함. row와 column들이 각각 같아질 때 loop가 끝나면 되겠다라고 생각했음. 그래서 while문을 쓰고 그 안에서 각자에 맞는 조건문을 넣은 loop를 더 씀. 어찌어찌 해결은 했는데 중간에 column 에서 역행으로 돌아가는 코드에서 엄청 버벅임. 

 

 

 

 

 

 

 

해결 코드

var diagonalSum = function(mat) {
    const sum = [];
    
    let startRow = 0;
    let endRow = mat.length - 1;
    let startCol = 0;
    let endCol = mat.length -1;
    
    while(startRow <= endRow && startCol <= endCol){
        for(let i = startRow; i<=endRow; i++){
            if(startRow === startCol || endRow === endCol){
                sum.push(mat[startRow][i])
            }
            
            startRow++
       }
    
         
        // row를 한번 초기화시켜줘야함.. 
        startRow = 0
        for(let i = endCol; i>=startCol; i--){
            // 중간에 겹쳐지는 row와 column값을 제거하기 위해서 
            if(endCol != startRow && endRow != startCol){
                console.log(endCol, endRow,startRow,)
                sum.push(mat[i][startRow])
                
            }
            endCol--
            startRow++
        }
        
        const add = sum.reduce((acc, cur) => acc += cur)
        console.log(sum, add)
        return add
        
        
    }
};

 

 

런타임 무슨일이야.. 처음엔 해결해서 마냥 좋았는데 다른 분들이 쓴 코드랑 비교해보고 조금 좌절함.. 

 

 

 

다른사람 코드

var diagonalSum = function(mat) {
    const len = mat.length;
    let sum = 0;
    
    mat.forEach((row, i) => {
        const l = i;
        const r = len - i - 1;
        sum += l === r ? row[l] : row[l] + row[r];
    });
    
    return sum;
};