[프로그래머스] 카펫

문제

Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 빨간색으로 칠해져 있고 모서리는 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

Leo는 집으로 돌아와서 아까 본 카펫의 빨간색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다.

Leo가 본 카펫에서 갈색 격자의 수 brown, 빨간색 격자의 수 red가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한사항

• 갈색 격자의 수 brown은 8 이상 5,000 이하인 자연수입니다.
• 빨간색 격자의 수 red는 1 이상 2,000,000 이하인 자연수입니다.
• 카펫의 가로 길이는 세로 길이와 같거나, 세로 길이보다 깁니다.

입출력 예

풀이

빨간색 격자의 수가 최소 1개니까 가로, 세로의 길이의 최솟값은 3이다. 그렇기 때문에 가로, 세로 길이의 최댓값은 전체 넓이인 brown + red를 최솟값인 3으로 나눈 값이 된다.

가로, 세로를 나타내는 x, y 변수를 3부터 (brown + red) / 3까지 탐색하며 조건을 만족하는 값을 찾는다.

이 때 카펫의 모서리를 두르고 있는 갈색 격자의 수인 2x + 2y - 4가 brown과 같고, 빨간색 격자의 수인 (x - 2) * (y - 2)가 red와 같으면 조건을 만족한다.

#include <vector>
using namespace std;

vector<int> solution(int brown, int red) {
    vector<int> answer;
    for (int y = 3; y <= (brown + red) / 3; y++) {
        for (int x = y; x <= (brown + red) / 3; x++) {
            if (2 * (x + y - 2) == brown && (x - 2) * (y - 2) == red) {
                answer.push_back(x);
                answer.push_back(y);
                return answer;
            }
        }
    }
    return answer;
}

출처: https://programmers.co.kr