[Lv. 0] 이차원 배열 대각선 순회하기

문제 출처 : https://school.programmers.co.kr/learn/courses/30/lessons/181829

프로그래머스

난이도 : Level.0

언어 : Kotlin

 

문제 설명 :

2차원 정수 배열 board와 정수 k가 주어집니다.

i + j <= k를 만족하는 모든 (i, j)에 대한 board[i][j]의 합을 return 하는 solution 함수를 완성해 주세요.

 

제한 사항 : 

• 1 ≤ board의 길이 ≤ 100
• 1 ≤ board[i]의 길이 ≤ 100
  • 1 ≤ board[i][j] ≤ 10,000
  • 모든 board[i]의 길이는 같습니다.
• 0 ≤ k < board의 길이 + board[i]의 길이

입출력 예 :

board	k	result
[[0, 1, 2], [1, 2, 3], [2, 3, 4], [3, 4, 5]]	2	8

풀이 : 

class Solution {
    fun solution(board: Array<IntArray>, k: Int): Int = board.indices.sumOf { i -> board[i].indices.filter { j -> i + j <= k }.sumOf { j -> board[i][j] } }
}

전에 풀었던 안전지대 문제를 재활용하여 쉽게 풀 수 있었다.