카펫

programmers.co.kr/learn/courses/30/lessons/42842

코딩테스트 연습 - 카펫

---

풀이

class Solution {
    public int[] solution(int brown, int yellow) {
        int tiles = brown + yellow;
        int[] answer = new int[2];
        for(int h = 3; h < brown; h++) {
            int w = tiles / h;
            if(yellow == (h - 2) * (w - 2)) {
                answer[0] = w;
                answer[1] = h;
                break;
            }
        }
        return answer;
    }
}

1. 총 타일 개수를 구한다. (variable: tiles)

 

2. 3부터 brown 전까지 반복한다. (variable: h)

    [문제 조건 상 brown >= 8이다. 즉 카펫의 한 변 >= 3이 성립된다.

    즉, 카펫의 한 변의 최소값은 3이라는 뜻이다. 따라서 반복문을 3부터 시작해도 되는 것이다.

    또한, 문제 조건에 보면 가로 >= 세로이다.

    그런데 반복문이 최소값에서부터 증가하기때문에 조건문에 맞는 경우(답을 찾은 경우), 반복문의 변수 h가 무조건 세로가 된다.]

 

    3. tiles / h를 하여 열을 구한다. (variable: w)

 

    4. (h -2) * (w - 2) == yellow이면

        [카펫에서 (행 - 2) * (열 - 2) = 노란 부분의 넓이가 된다. 그림으로 생각하면 쉽다.]

 

        5. answer[0]에 w, answer[1]에 h 대입 후 반복문을 끝낸다.

 

6. answer를 return한다.