프로그래머스 - 카펫 (레벨2, swift)

문제 출처

 

programmers.co.kr/learn/courses/30/lessons/42842

코딩테스트 연습 - 카펫

 

문제 설명

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

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) 먼저 노란색 격자의 수에 따라 만들 수 있는 노란색 모양의 경우의 수를 찾아야 한다.

yellow가 2라면 [2X1] 이렇게 1가지 경우가 있다. 가로가 세로보다 같거니 길기 때문에 [1X2]는 될 수 없다.

yellow가 1이라면 [1X1] 1가지 경우가 있다.

yellow가 24라면 [24X1], [12X2], [8X3], [6X4] 이렇게 4가지 경우가 있다.

 

-> 주어진 yellow의 가로 값(width)은 세로(height)보다 같거나 크기 때문에 제곱근부터 시작해 1씩 더해준다. width값에 맞춰 height값도 1부터 시작해 1씩 더해주며 주어진 노란색 격자 수에 맞는 값을 찾아주면 된다.

 

 

2) 노란색 모양의 크기에 따라 갈색 카펫의 크기를 계산하고 주어진 격자의 수와 일치하는 지 확인한다.

만약 yellow가 [2X1]이라면 갈색의 크기는

 

  [1] yellow의 가로 크기보다 2씩 더 큰 값이 위, 아래로 있고(하얀색으로 색칠된 부분)

 

  [2] yellow의 세로 크기만큼이 좌, 우로 있게 된다(하얀색으로 색칠된 부분)

 

-> yellow의 크기가 [2X1]이라면 갈색 부분은 (2+2) X 2 + 1 X 2가 된다. 이 값을 주어진 갈색 격자수와 비교해 보고 맞은 지 체크해 보고 맞다면 카펫의 크기를 계산해서 반환해주면 된다. yellow의 크기에 각각 2를 더해준 [2+2, 1+2]의 값인 [4, 3]을 return 해주면 된다.

 

 

위 과정을 코드로 구현하면 아래와 같다. 자세한 내용은 주석을 참고하면 된다.