[프로그래머스 Level 2] 카펫

🖊️ 문제

카펫  

 

Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 노란색으로 칠해져 있고 테두리 1줄은 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다.

 

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

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

 

제한사항

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

 

🖥️ 코드

#include <string>
#include <vector>
#include <math.h>

using namespace std;

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

    for (int height = 3; height <= sqrt(sum); height++)
        if (!(sum % height))
            if ((height - 2) * ((sum / height) - 2) == yellow)
                return { sum / height, height };
}

 

🤔 나의 생각

다른 사람들이 푼 문제를 보니 연립 방정식으로 풀었던데... 고등학교를 갓 졸업했다면 나도 저렇게 풀었을 것 같다.

 

헷갈리지 말아야 할 건 가장자리 색은 무적권 갈색이고 가장자리를 제외한 카펫의 색은 노란색이다. 가장자리를 제외한 격자에 갈색과 노란색이 섞여있을 순 없다!

 

카펫의 높이는 최소 3으로 시작한다. 전체 격자의 수를 높이로 나눴을 때 나머지가 0이라면 높이*몫으로 카펫을 배치할 수 있다. 여기서 노란색 격자의 수는 (높이 - 2) * (가로 - 2)으로 그 수가 문제 조건의 yellow의 수와 동일하다면 정답이다.