[프로그래머스] Lv.2 카펫 JavaScript

문제 설명

Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

!https://grepp-programmers.s3.ap-northeast-2.amazonaws.com/files/production/b1ebb809-f333-4df2-bc81-02682900dc2d/carpet.png

Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.

Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한사항

• 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
• 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
• 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

입출력 예

brown	yellow	return
10	2	[4, 3]
8	1	[3, 3]
24	24	[8, 6]

---

문제풀이

1. total 변수를 이용하여 카펫의 가로와 세로의 합을 구한다.
2. width * height = total이어야 한다.
3. 가능한 카펫의 가로와 세로 길이를 찾기 위해 for 반복문을 사용하는데, height 변수를 3부터 시작하여 total의 제곱근까지 증가시킨다.
   • 3부터 시작하는 이유: 노란색 카펫이 최소 1개이상 있어야 하고, 갈색 카펫이 노란색을 감싸려면 최소한의 세로 길이는 3이다.
4. width 변수를 계산하여 카펫의 가로 길이를 구한다. total을 height로 나눈 값을 소수점 이하를 버린 정수로 할당한다.
5. 이후, if 문을 사용하여 노란색 카펫의 개수를 검사한다. 노란색 카펫은 갈색 카펫의 가장자리(모서리 부분)에 위치하지 않으므로, 가로와 세로 길이에서 각각 2를 뺀 후 곱한 값이 노란색 카펫의 개수와 일치한다면 [가로, 세로]값을 반환한다.

✅ 답안

function solution(brown, yellow) {
  // 카펫의 가로와 세로의 합을 구함
  const total = brown + yellow;

  // 가능한 카펫의 가로와 세로의 길이를 찾음
  for (let height = 3; height <= Math.sqrt(total); height++) {
    const width = Math.floor(total / height);

    // 노란색 카펫의 개수를 검사하여 정답을 찾음
    if ((width - 2) * (height - 2) === yellow) {
      return [width, height];
    }
  }

  return [];
}