[lv.2] 카펫

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

• 문제

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

• 풀이
  1) 전체 칸의 수 = brown+yellow
  2) 위의 칸을 가진 사각형을 a라고 할때, 가로 i와 세로 j이고, i*j=a인 경우를 찾는다.
  3) 찾을 경우 yellow칸은 (i-2)x(j-2)이다. (brown이 1줄로 감싸져서)
  위 두 값이 서로 맞으면 그게 답.

• 코드

class Solution {
    public int[] solution(int brown, int yellow) {
        int[] result = new int[2];
        int total = brown+yellow;
        for(int i=total/2; i>0; i--){
            if(total%i==0){
                int j = total/i;
                if((i-2)*(j-2)==yellow){
                    result[0] = i;
                    result[1] = j;
                    break;
                }
            }
        }
        return result;
    }
}