[백준 | Javascript] 2775

기본 수학 1. 6단계
2775번. 부녀회장이 될테야

문제

평소 반상회에 참석하는 것을 좋아하는 주희는 이번 기회에 부녀회장이 되고 싶어 각 층의 사람들을 불러 모아 반상회를 주최하려고 한다.
이 아파트에 거주를 하려면 조건이 있는데, “a층의 b호에 살려면 자신의 아래(a-1)층의 1호부터 b호까지 사람들의 수의 합만큼 사람들을 데려와 살아야 한다” 는 계약 조항을 꼭 지키고 들어와야 한다.
아파트에 비어있는 집은 없고 모든 거주민들이 이 계약 조건을 지키고 왔다고 가정했을 때, 주어지는 양의 정수 k와 n에 대해 k층에 n호에는 몇 명이 살고 있는지 출력하라. 단, 아파트에는 0층부터 있고 각층에는 1호부터 있으며, 0층의 i호에는 i명이 산다.

입력

첫 번째 줄에 Test case의 수 T가 주어진다. 그리고 각각의 케이스마다 입력으로 첫 번째 줄에 정수 k, 두 번째 줄에 정수 n이 주어진다

출력

각각의 Test case에 대해서 해당 집에 거주민 수를 출력하라.

제한

1 ≤ k, n ≤ 14

예제 입출력

예제 입력

2
1
3
2
3

예제 출력

6
10

solution

const fs = require("fs");
const input = fs.readFileSync("/dev/stdin").toString().split("\n");

const iter = Number(input[0]);

// 0층의 i호에는 i명이 산다
// floor층의 room호에 살기 위한 인원 조건은
// (floor-1)층의 1호부터 room호까지의 인원을 합한 인원만큼 살아야한다는 것이다.
// 1층 3호에 살려면, 0층 1,2,3호 인원을 합친 인원이 필요하므로
// 1+2+3 = 6명이 1층 3호에 살기 위한 조건이 된다.

for(let i = 1; i < iter+1; i++){
  const floor = Number(input[2*i - 1]);
  const room = Number(input[2*i]);
 
  // 층, 호수에 대한 인원수 정보를 저장하기 위한 2차원 배열 생성
  // eachFloor[floor][room]
  // 0으로 채워짐
  // Array(1)로 하면 0층 데이터만 넣을 수 있으므로
  // floor층까지의 데이터를 다루고자 한다면 Array(floor + 1) 사용
  // room은 1호부터 시작하므로 Array(room)을 사용해도 문제 없음
  const eachFloor = Array.from(Array(floor + 1), () => Array(room).fill(0));
  
  // 필요한 호실까지 0층 데이터 생성
  for (let i = 1; i <= room; i++) {
    eachFloor[0][i] = i;
  }
  
  // 1 ~ floor층
  // floor층 room호에는 (floor-1층 room호 + floor층 room-1호)만큼 산다
  for (let i = 1; i <= floor; i++) {
    for (let j = 1; j <= room; j++) {
      eachFloor[i][j] = eachFloor[i - 1][j] + eachFloor[i][j-1];
    }
  }
  
  // floor층 room호 필요 인원수 출력
  console.log(eachFloor[floor][room]);
}

해설

2차원 배열을 사용한 것까지는 좋았는데, 인원수 계산하는 법에서 꼬여버려서 시간을 너무 끌게되었다. 엄청 비슷하게 접근하신 분이 계셔서 풀이를 참고해서 진행했다.
from() 메서드를 통해 2차원 배열을 생성하는 법을 알게 되었다. 아직까지는 잘 이해가 되지않지만, 유용하게 쓰일 것 같다.
풀이 참고
Array.from()으로 2차원 배열 만들기