► JS Algorithm/Programmers

[Programmers] JavaScript 알고리즘 | Lv.0 이차원 배열 대각선 순회하기

다람트리 2024. 7. 21. 19:01
반응형

🔒 문제 설명

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 of mine

단항 더하기 (+)

function solution(board, k) {
  let answer = 0;
  for (let i in board) {
    for (let j in board[0]) {
      if (+i + +j <= k) {
        answer += board[i][j];
      }
    }
  }
  return console.log(answer);
}

solution(
  [
    [0, 1, 2],
    [1, 2, 3],
    [2, 3, 4],
    [3, 4, 5],
  ],
  2
); //output: 8

 


 

🔐 solution of others

Array.prototype.reduce()

const solution = (board, k) => {
  return console.log(
    board.reduce(
      (total, row, i) =>
        total +
        row.reduce((prev, num, j) => (i + j <= k ? prev + num : prev), 0),
      0
    )
  );
};

solution(
  [
    [0, 1, 2],
    [1, 2, 3],
    [2, 3, 4],
    [3, 4, 5],
  ],
  2
); //output: 8

 


 

반응형