[programmers] 카펫 - Java

카펫

문제 설명

제한사항

• 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.

• 노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.

• 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

입출력 예

solution.java

class Solution {
    public int[] solution(int brown, int yellow) {
        int total = brown + yellow;
        int[] result = new int[2];
            
        // 제곱근까지 반복하면 중복 없이 계산 가능
        for (int i = 1; i <= Math.sqrt(total); i++) {
            // 나머지가 0인 경우 직사각형
            if (total % i == 0) {
                int row = total / i;
                int col = i;
                //가로와 세로에서 1씩 뺀 값이 노란색과 일치할 경우
                if ((row - 2) * (col - 2) == yellow) {
                    result[0] = row;
                    result[1] = col;
                    break;
                }
            }
        }
        
        return result;
    }
}

핵심 키워드

• 제곱근까지만 반복하는 것은, 가로와 세로 길이가 같은 경우를 고려하기 위함이다. 제곱근 이상의 숫자에 대해서는 이미 확인한 경우의 반대편 길이와 동일한 결과를 가지므로 추가적인 반복은 필요하지 않다.

결론!