[프로그래머스] 카펫

https://programmers.co.kr/learn/courses/30/lessons/42842?language=java 

코딩테스트 연습 - 카펫

 

분류는 탐색인데 수학으로 접근이 돼서 연립방정식으로 풀었다.

 

카펫의 가로의 길이를 n, 세로의 길이를 m 이라 하자.

그러면 1) n + m = n * m (brown + yellow) - ((n-2) * (m-2) (yellow)) / 2 + 2 로 나타낼 수 있다.

2) n * m 은 brown + yellow 로 1) 과 2) 를 연립하면 (n + m) ^ 2 - 4 * n * m = (n - m)^2 로 n 과 m 을 구할 수 있다. 

 

brown 은 테두리 1줄을 뜻하므로 yellow 가 (n-2) * (m-2) 가 된다.

 

class Solution {
    public int[] solution(int brown, int yellow) {
        int[] answer = new int[2];
        int x = (brown / 2) + 2;
        int y = brown + yellow;
        answer[0] = (x + (int) Math.sqrt(x * x - (4 * y))) / 2;
        answer[1] = x - answer[0];
        
        return answer;
    }
}