[프로그래머스] 카펫

1. 문제

원본 링크 : https://school.programmers.co.kr/learn/courses/30/lessons/42842

 

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

 

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

 

2. 코드

#include <string>
#include <vector>

using namespace std;

vector<int> solution(int brown, int yellow) {
    vector<int> answer;
    
    //완전 탐색
    int sum = brown+yellow;
    for(int i=2;i<sum;i++){
        if(sum%i!=0) continue;
        else
        {
            if((i-2)*(sum/i-2)==yellow)
            {
                answer.push_back(sum/i);
                answer.push_back(i);
                break;
            }
            else continue;
        }
    }
    
    return answer;
}

 

3. 해설

//완전 탐색
    int sum = brown+yellow;
    for(int i=2;i<sum;i++){
        if(sum%i!=0) continue;

먼저 brown+yellow를 더하면 모든 타일의 갯수를 구할 수 있습니다.

타일의 갯수는 가로x세로의 경우의 수를 구할 수 있기에 구해줍니다.

타일의 세로 길이는 1보다 커야하기에(0이나 1인 경우 노란색 구역이 생길 수 없음) i=2부터 for문을 돌려줍니다.

if(sum%i!=0) : 만약 나누어 질수 없는 경우라면 가로 세로를 구할수 없기 때문에 계산하지 않고 넘어갑니다.

        else
        {
            if((i-2)*(sum/i-2)==yellow)
            {
                answer.push_back(sum/i);
                answer.push_back(i);
                break;
            }
            else continue;
        }
    }

나누어지는 경우라면, 해당 가로 세로의 길이에서 2씩 빼서 yellow 영역의 갯수를 구합니다.

만약 yellow 갯수가 맞다면 해당 값들을 answer에 저장해줍니다.