[JAVA] △ 카펫

문제 설명
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];
        // yellow 가 m * n으로 이루워져있는지 파악
        // 해당 (m+2)*2+(n+2)*2-4 가 brown과 같으면 됨
        for(int i=1; i<Math.sqrt(2000000); i++){
            if(yellow%i==0){
                int m = i;
                int n = yellow/i;
                int sum = (m+2)*2 + (n+2)*2 -4;
                if(brown==sum){
                    answer[0] = Math.max(m+2, n+2);
                    answer[1] = Math.min(m+2, n+2);
                    return answer;
                }
            }
        }
        return answer;
    }
}

풀이 2

class Solution {
    public int[] solution(int brown, int yellow) {
        int a = (brown+4)/2; // 전체 가로+세로 합
        int b = yellow+brown; // 주어진 타일 합 
        int max = (int)(a+Math.sqrt(a*a-4*b))/2;
        int min = (int)(a-Math.sqrt(a*a-4*b))/2;
        int[] answer = {max,min};
        return answer;
    }
}