프로그래머스 카펫 (Level2 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]

---

문제 해결을 위한 과정

1. 전체 넓이 구하기: brown + yellow는 카펫의 전체 면적(width * height)입니다.
2. 가로, 세로 후보군 탐색: 가로 길이는 세로 길이와 같거나 길기때문에, 가로(width)를 큰 값부터 줄여나가며 탐색합니다.
3. 노란색 격자의 법칙: 갈색 테두리가 한 줄씩 감싸고 있으므로, 내부 노란색 격자의 가로는 width - 2, 세로는 height - 2가 됩니다.
4. 조건 검증: (width - 2) * (height - 2)의 결과가 주어진 yellow같다면 정답입니다.
   
   

---

소스코드

import java.util.*;

class Solution {
    public int[] solution(int brown, int yellow) {
        int[] answer = new int[2];
        
        int width = 0;
        int height = 0;
        
        for(int i = brown + yellow; i >= 1; i--) {
            width = i;
            height = (brown + yellow) / width;
            
            if((width - 2) * (height - 2) == yellow) {
                break;
            }
        }
        
        answer[0] = width;
        answer[1] = height;
        
        return answer;
    }
}