[JS] 프로그래머스 - 카펫

📌  문제

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

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

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

제한사항

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

 

💡  아이디어

'가로 = 노랑가로+2', '세로 = (갈색 수+노란색 수)/가로' 이다.

노란색을 어떻게 배치하느냐에 따라서 카펫의 가로, 세로 크기가 달라지므로 일단 노랑의 약수를 구한다.

약수 중에서 긴 것 부터 하나씩 노랑의 세로로 두고, 탐색한다.

'가로 * 세로 = 갈색 수 + 노란색 수' 이므로 이 조건을 만족하는 가로 세로를 찾는다.

 

✏️  풀이

function solution(brown, yellow) {
  var answer = [];
  let divY = [];
  let X = 0;
  let Y = 0;
  for(let i=1; i<=Math.sqrt(yellow); i++){
  // 약수 찾기
    if(yellow % i === 0) {
      divY.push(i);
      if(yellow / i != i) divY.push(yellow / i);
    }
  }
  divY.sort((a, b) => b - a);
  // 내림차순으로 정렬

  for(let i=0; i<divY.length; i++){
  // 약수 중에서 하나씩 노랑의 가로, 세로로 두고 총 카펫의 가로 세로 구하기
    X = divY[i] + 2;
    if(X < brown && (brown + yellow) % X === 0){
      Y = (brown + yellow) / X;
    }
    if(X * Y === brown + yellow) {
      answer.push(X);
      answer.push(Y);
      break; 
    }
  }
  return answer;
}