이차원 배열 대각선 순회하기

도비김·2024년 2월 27일
0

Programmers | lv.0 기초 

목록 보기
124/125
문제 설명

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

입출력 예 설명

입출력 예 #1

  • 입출력 예 #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 합니다.

solution

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,
  );
}
profile
To Infinity, and Beyond!

0개의 댓글