[프로그래머스] 카펫 (완탐)

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 sum = brown + yellow; //갈색, 노란색 총 개수 더하면 (ex. 8+1  = 9 -> 3*3 = 9)
        //대신 가로가 더 길게.
        //조건 추가: 세로가 짝수이면 안됨 (왜냐하면 가운데에 노란색이 있어야하는데 짝수가 되는 순간 가운데 자리가 없음)
        //또 조건이 있는데 이게 생각이 안난다...
        for(int width = sum; width>=1; width--){
            if(sum % width == 0){
            
                int height = sum / width;  // 해당 width에 대한 height

                if(width >= height){
                    //가로가 세로보다 클 때 
                
                    if((width -2)* (height-2) == yellow){
                         answer[0] = width;
                         answer[1] = height;
                      break;
                    }
                }
            } 
        }
        
        return answer;
    }
}