[Algorithm /프로그래머스] 카펫

[문제 설명]
Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

carpet.png

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]

 

[문제 해결]

import java.util.*;
class Solution {
    public int[] solution(int brown, int yellow) {
        int[] answer = new int[2];
        
        int result= brown + yellow; //전체 넓이 : yellow + brown
        int width = 0,  height = 0; 
        
        for(int i = 3 ; i <= result; i++){ 
            if( result % i == 0){
                int param = result/ i; //넓이 구해지는 것이 가로 또는 세로
                
                if(param >= 3 && i >= 3) //가로 세로 최소길이 3부터 시작
                {
                    int cal = (param-2)*(i-2); //brown이 yellow를 감싸므로, yellow는 width-2, height*2 곱셈
                    if(cal == yellow){
                        width = Math.max(param, i); //가로가 길다는 조건이 있으므로, Max인것이 가로, Min인 것이 세로
                        height = Math.min(param,i); 
                    }  
                }
            }
        }
        
        answer[0] = width;
        answer[1] = height;
      
        return answer;
    }
}