카펫 - 프로그래머스

문제

Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.
Leo는 집으로 돌아와서 아까 본 카펫의 노란색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.
Leo가 본 카펫에서 갈색 격자의 수 brown, 노란색 격자의 수 yellow가 매개변수로 주어질 때
카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한사항
갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
노란색 격자의 수 yellow는 1 이상 2,000,000 이하인 자연수입니다.
카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

문제 풀이

원리만 이해하면 되는 완전 탐색 문제 유형.

리턴되는 배열의 곱이 두 격자의 합이고 결국 배열 값을 두 격자의 합에 약수 중에서 조건을 만족하는 배열을 리턴하면 됐다.

1. 각 격자의 수는 노란 격자를 포함하고 모서리 길이가 1이어야 하기 때문에 격자는 3 이상이 돼야 한다.
2. 격자의 합을 탐색하며 합의 약수이면서 3 이상의 두 짝을 구한 뒤 가운데 사각형 넓이를 구한다.
3. 가운데 사각형과 노란색 넓이가 같으면 두 가로 세로를 뺀다.
4. 가로 길이가 세로 길이 이상이어야 한다고 했으니 가로, 세로 순으로 배열에 넣었다.

class Carpet {
  public int[] carpet(int brown, int yellow) {
    int[] answer = new int[2];
    int sum = brown + yellow;
    
    for (int i = 3; i < sum; i++) {
    
      int remainder = sum / i;
      
      if (sum % i == 0 && remainder >= 3) {
        int width = Math.max(i, remainder);
        int height = Math.min(i, remainder);
        int middle = (height - 2) * (width - 2);
        
        if (middle == yellow) return new int[] {width, height};

      }
      
    }

    return new int[]{0, 0};
  }
}