[프로그래머스] Lv.2 : 카펫 - 자바[Java]

https://school.programmers.co.kr/learn/courses/30/lessons/42842

프로그래머스

풀이

노란색 격자의 가로/세로 길이가 될 수 있는 모든 경우를 구한 다음, (노란색 가로 + 2) * (노란색 세로 + 2) - 노란색 개수 = 갈색 개수 가 되는 경우를 찾아서 답을 구한다.

코드

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