문제
Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.
Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.
Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.
제한사항
- 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
- 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
- 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.
- 출력 예시 -
정답 코드
< 내 정답 코드 >
class Solution {
public int[] solution(int brown, int yellow) {
int[] answer = new int[2];
int size = brown + yellow;
for (int i = 3; i <= size; i++) {
int row = i;
int col = size / row;
if (row < col) {
continue;
}
if ((row - 2) * (col - 2) == yellow) {
answer[0] = row;
answer[1] = col;
}
}
return answer;
}
}
이것을 주의하자!
- 수학적으로 생각하는데 어려움을 겪었던 문제다. 해당 문제에 등장하는 특징을 파악해야 그들을 조합해서 풀 수 있다.
- 일단 yellow 와 brown 의 개수를 알면 '전체 타일 수' 를 알 수 있다. 사각형 넓이의 공식이 " 가로 X 세로 " 이기 때문에 전체 크기의 약수들의 조합이야말로 가능한 가로, 세로 길이가 된다. 따라서 for 문을 이용해 가로 , 세로 경우의 수를 하나씩 계산한다. 여기서 만약 yellow 가 1개일 때 brown 의 가로와 세로 최소 길이는 3이다. 따라서 for 문에서 시작을 3부터 해도 된다.
- 가로, 세로 경우의 수를 구하면, 해당 값이 조건과 일치하는지 본다. 먼저 "가로의 길이는 세로보다 작을 수 없다". 그래서 if 문으로 가로의 길이가 세로보다 작으면 continue 키워드를 통해 돌려 보낸다.
- 중요한 특징 중 하나가 바로 'yellow 타일 개수' 구하는 공식이다. 생각해보면 yellow 타일의 개수는 전체 타일의 가로 개수보다 2개 적고, 세로 길이보다 2개 적다. 따라서 ' yellow 개수 = (전체 가로 길이 - 2) * (전체 세로 길이 - 2) ' 가 된다. 위에서 계산된 가로, 세로 경우의 수를 이 공식에 대입해서 yellow 개수가 맞는 값이 정답이 되는 것이다.
'코딩 테스트' 카테고리의 다른 글
[프로그래머스] (Lv.2) 구명보트 *** (0) | 2022.09.13 |
---|---|
[프로그래머스] (Lv.2) 짝지어 제거하기 ** (1) | 2022.09.13 |
[프로그래머스] (Lv.2) 영어 끝말잇기 * (0) | 2022.09.13 |
[프로그래머스] (Lv.2) 다음 큰 숫자 (0) | 2022.09.13 |
[프로그래머스] (Lv.1) 모의고사 *** (0) | 2022.09.11 |