본문 바로가기

코딩 테스트

[프로그래머스] (Lv.2) 카펫 ***

728x90

문제


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 의 개수를 알면 '전체 타일 수' 를 알 수 있다. 사각형 넓이의 공식이 " 가로  세로 " 이기 때문에 전체 크기의 약수들의 조합이야말로 가능한 가로, 세로 길이가 된다. 따라서 for 문을 이용해 가로 , 세로 경우의 수를 하나씩 계산한다. 여기서 만약 yellow 가 1개일 때 brown 의 가로와 세로 최소 길이는 3이다. 따라서 for 문에서 시작을 3부터 해도 된다.

 

-  가로,  세로 경우의 수를 구하면, 해당 값이 조건과 일치하는지 본다. 먼저 "가로의 길이는 세로보다 작을 수 없다". 그래서 if 문으로 가로의 길이가 세로보다 작으면 continue 키워드를 통해 돌려 보낸다.

 

-  중요한 특징 중 하나가 바로 'yellow 타일 개수' 구하는 공식이다. 생각해보면 yellow 타일의 개수는 전체 타일의 가로 개수보다 2개 적고, 세로 길이보다 2개 적다. 따라서 ' yellow 개수  =  (전체 가로 길이 - 2)  *  (전체 세로 길이 - 2) ' 가 된다. 위에서 계산된 가로,  세로 경우의 수를 이 공식에 대입해서 yellow 개수가 맞는 값이 정답이 되는 것이다.