2차원 정수 배열 board
와 정수 k
가 주어집니다.
i
+ j
<= k
를 만족하는 모든 (i
, j
)에 대한 board[i][j]
의 합을 return 하는 solution 함수를 완성해 주세요.
board | k | result |
---|---|---|
[[0, 1, 2],[1, 2, 3],[2, 3, 4],[3, 4, 5]] | 2 | 8 |
입출력 예 #1
board
를 표로 나타내면 다음과 같습니다.i \ j | 0 | 1 | 2 |
---|---|---|---|
0 | 0 | 1 | 2 |
1 | 1 | 2 | 3 |
2 | 2 | 3 | 4 |
3 | 3 | 4 | 5 |
i
+ j
가 2보다 작거나 같은 항들의 합은 0 + 1 + 2 + 1 + 2 + 2 = 8이므로 8을 return 합니다.function solution(board, k) {
return board.map((x,i)=>x.filter((y,j)=>i+j<=k)).flat().reduce((a,c)=>a+c)
}
다른풀이
function solution(board, k) {
return board.reduce(
(total, row, i) => total + row.reduce((prev, num, j) => (i + j <= k ? prev + num : prev), 0),
0,
);
}