프로그래머스 카펫 C++

문제 링크 : 

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

프로그래머스

 

카펫의 가로 길이를 x, 세로 길이를 y라고 하자.

x * y = brown + yellow
(x-2) * (y-2) = yellow
x>=y

문제에서 보면 위의 세 조건을 만족시켜야만한다.

그래서 각각의 조건을 만족시키는 x, y값을 for문을 통해서 x 값을 하나씩 증가시키면서 확인해보면 된다.

 

정답 코드는 아래와 같다

 

#include <string>
#include <vector>

using namespace std;

vector<int> solution(int brown, int yellow) {
    vector<int> answer;
    int x, y;
    int total_size = brown+yellow;
    for (int x=1; x<=brown;x++){
        if ((total_size % x==0 && total_size/x!=0)){
            y = total_size/x;
            if ((x-2) * (y-2) == yellow && x>=y){
                answer.push_back(x);
                answer.push_back(y);
            }
        }
    }
    return answer;
}