[프로그래머스] 카펫

문제 :

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

프로그래머스

 

해결 :

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

 

전체 격자 수의 약수를 구하고, 각 약수에 대해 가로와 세로를 계산하여 조건을 만족하는지 확인한다.