[자바] 프로그래머스 카펫

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

프로그래머스

 

완전탐색 문제였다. 문제 자체가 어렵다기 보다는 마지막에 답을 확인하는 조건이 약간 어려웠다. 일단 brown+yellow로 나온 수의 약수 중에 return 해야되는 값이 있는 건 알고 있었는데, 단순히 약수 중에 제일 제곱수에 가까운 약수 쌍을 구하면 되는 줄 알았는데 아니었다..

 

잘 모르겠어서 찾아보니 완전 당연한 조건을 생각하지 못하고 있었음을 알게 되었다.

 

1. 약수 쌍에 해당하는 두 약수 모두 3이상이어야 함 => 내부에 yellow를 하나라도 갖고 있게 된다면 yellow를 감싸는 brown의 가로 세로 길이는 모두 위아래 하나씩과 yellow 하나, 그래서 3 이상의 값이어야 한다.

2. 만들어지는 전체 사각형의 세로 - 2 * 사각형의 가로 - 2 는 yellow 의 값과 동일해야함 => 1번 조건과 마찬가지로 yellow를 감싸는 brown 양쪽의 가로 세로 하나씩 빼면 총 2를 빼는 거니까..

(EX) yellow = 2 라면, (4-2)*(3-2) = 2, yellow = 24 라면, (8-2) * (6-2) = 24

 

그래서 구현한 방법은 아래와 같음

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

answer[0]에 더 큰 수를 넣어야 하는데 나는 for문을 제곱수까지만 돌리게 했으니까, 무조건 i보다는 sum/i 값이 클 수 밖에 없어서 바로 집어넣었다.