본문 바로가기

Archived(CSE Programming)/알고리즘(C++)

프로그래머스_카펫

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

 

코딩테스트 연습 - 카펫 | 프로그래머스

Leo는 카펫을 사러 갔다가 아래 그림과 같이 중앙에는 빨간색으로 칠해져 있고 모서리는 갈색으로 칠해져 있는 격자 모양 카펫을 봤습니다. Leo는 집으로 돌아와서 아까 본 카펫의 빨간색과 갈색으로 색칠된 격자의 개수는 기억했지만, 전체 카펫의 크기는 기억하지 못했습니다. Leo가 본 카펫에서 갈색 격자의 수 brown, 빨간색 격자의 수 red가 매개변수로 주어질 때 카펫의 가로, 세로 크기를 순서대로 배열에 담아 return 하도록 solution 함수

programmers.co.kr

해당 문제는 완전탐색으로 모든 경우를 다 살펴보는 문제이다.

먼저 brown과 red 카펫의 수를 더해 전체 너비를 구하고 해당 너비로 구현가능한 직사각형의 후보들을 전부 고려한다. 그리고 해당 후보들 중에서 둘레의 길이로 brown 카펫의 수를 커버 가능한 지를 확인하면 된다.

#include <string>
#include <vector>

using namespace std;

vector<int> solution(int brown, int red) {
    vector<int> answer;
    int br = brown+red; // 전체 너비
    vector<pair<int,int>> pv; // 가능한 후보들 추가
    for(int i=2;i*i<=br;i++){
        if(br % i ==0){
            pv.push_back(pair<int,int>(br/i,i)); // 후보 추가
        }
    }
    
    // 후보들 중 둘레의 길이만큼 brown으로 감쌀 수 있는 지
    for(int i=0;i<pv.size();i++){
        if(((pv[i].first *2 )+ (pv[i].second *2) - 4) == brown){
            answer.push_back(pv[i].first);
            answer.push_back(pv[i].second);
        }
    }
    
    return answer;
}