[Algorithm] 프로그래머스 - 이차원 배열 대각선 순회하기 (Kotlin)

 

조건에 맞춰서 합을 구한 다음 리턴하면 된다.

 

class Solution {
    fun solution(board: Array<IntArray>, k: Int): Int {
        var sum = 0

        for (i in board.indices) {
            for (j in board[i].indices) {
                if (i + j <= k) {
                    sum += board[i][j]
                }
            }
        }

        return sum
    }
}