[프로그래머스] 카펫 | C++

카펫

접근 방식

전체 카펫의 크기는 brown과 yellow를 더한 것과 같다.

int total = brown + yellow;

이제 우리가 구하고자 하는
전체 카펫 크기의 가로를 W, 세로를 H로 정의해보자.

이 때 brown은 전체 카펫의 크기 W × H에서 yellow의 갯수를 뺀 것과 같고,
yellow는 (W - 2) × (H - 2)와 같다.

이를 정리하면,
brown = 2H + 2W - 4
yellow = WH - 2H - 2W + 4이다.

이를 충족하는 W와 H를 for문으로 탐색해서 찾아보았다.

전체 소스코드

#include <vector>
using namespace std;

vector<int> solution(int brown, int yellow) {
    vector<int> answer;

    int total = brown + yellow;

    for(int H = 1; H*H <= total; ++H) {
        if(total%H == 0) {
            int W = total/H;

            if( (brown == (2*H + 2*W - 4)) && (yellow == (W*H - 2*H - 2*W + 4)) )
            {
                answer.push_back(W);
                answer.push_back(H);
            }
        }
    }

    return answer;
}