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

카펫 문제
https://school.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]

---

내 코드

class Solution {
    public int[] solution(int brown, int yellow) {
        int[] answer = new int[2]; // [가로, 세로]

        for(int i = 1; i <= yellow; i++){
            int yellow_width = Math.max(i, yellow/i);
            int yellow_height = Math.min(i, yellow/i);

            int answer_width = yellow_width + 2;
            int answer_height = yellow_height + 2;
		
            if((answer_width + answer_height) * 2 - 4 == brown && answer_height * answer_width == brown + yellow ){
                answer[0] = answer_width;
                answer[1] = answer_height;
                break;
            }
            
        }
        return answer;
    }
}

• 우선적으로 yellow의 가로와 세로의 격자의 개수를 알기위해 yellow / i를 해주었고 제한 사항 중에 가로가 세로보다 크거나 같다고 했기에 Math.max()를 이용하였다.
• answer_width와 answer_heigth는 양 끝 모서리를 포함하기 때문에 둘 다 +2를 해주었다.
• brwon + yellow == 카펫의 가로 * 카펫의 세로이기 때문에 if 문의 넣어줬고, yellow의 가로 세로 격자의 수 따라 brown의 격자의 수가 달라지기 때문에 이도 조건에 넣어주었다.

알게된 점

• 문제 예시에서 주어진 값들을 보다 brwon + yellow == 카펫의 가로 * 카펫의 세로라는 식을 도출하여 문제를 풀 수 있었다.
• 어떠한 문제를 풀 때 해당 문제를 풀기위해 식을 만들어야할 때가 종종 있다. (저번에 풀었던 전화번호 누르기도 식을 따로 짰어야 했다.) (식 짜는 것을 무서워하지 말자!)