1-3. 최소 직사각형( vector 정렬, 값 교환 )

1. 문제

명함 지갑을 만드는 회사에서 지갑의 크기를 정하려고 합니다. 다양한 모양과 크기의 명함들을 모두 수납할 수 있으면서, 작아서 들고 다니기 편한 지갑을 만들어야 합니다. 이러한 요건을 만족하는 지갑을 만들기 위해 디자인팀은 모든 명함의 가로 길이와 세로 길이를 조사했습니다.

아래 표는 4가지 명함의 가로 길이와 세로 길이를 나타냅니다.

명함 번호	가로 길이	세로 길이
1	60	50
2	30	70
3	60	30
4	80	40

가장 긴 가로 길이와 세로 길이가 각각 80, 70이기 때문에 80(가로) x 70(세로) 크기의 지갑을 만들면 모든 명함들을 수납할 수 있습니다. 하지만 2번 명함을 가로로 눕혀 수납한다면 80(가로) x 50(세로) 크기의 지갑으로 모든 명함들을 수납할 수 있습니다. 이때의 지갑 크기는 4000(=80 x 50)입니다.

모든 명함의 가로 길이와 세로 길이를 나타내는 2차원 배열 sizes가 매개변수로 주어집니다. 모든 명함을 수납할 수 있는 가장 작은 지갑을 만들 때, 지갑의 크기를 return 하도록 solution 함수를 완성해주세요.

---

제한사항

• sizes의 길이는 1 이상 10,000 이하입니다.
  • sizes의 원소는 [w, h] 형식입니다.
  • w는 명함의 가로 길이를 나타냅니다.
  • h는 명함의 세로 길이를 나타냅니다.
  • w와 h는 1 이상 1,000 이하인 자연수입니다.

---

입출력 예

[[60, 50], [30, 70], [60, 30], [80, 40]]	4000
[[10, 7], [12, 3], [8, 15], [14, 7], [5, 15]]	120
[[14, 4], [19, 6], [6, 16], [18, 7], [7, 11]]	133

 

2. 코드

#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int cmp(vector<int>& vec1, vector<int>& vec2)
{
    return vec1 > vec2;
}


int cmp2(vector<int>& vec1, vector<int>& vec2)
{
    return vec1[1] > vec2[1];
}


int solution(vector<vector<int>> sizes) {
    int temp = 0;
    int best1 = 0;
    int best2 = 0;

    for (auto& num : sizes)
    {
        if (num[0] < num[1])
        {
            temp = num[0];
            num[0] = num[1];
            num[1] = temp;
        }
    }

    sort(sizes.begin(), sizes.end(), cmp);
    best1 = sizes[0][0];

    sort(sizes.begin(), sizes.end(), cmp2);
    best2 = sizes[0][1];
    
    return best1 * best2;
}

 

3. 분석

주어진 모든 명함을 넣을 수 있는 최소 크기의 직사각형 사이즈를 구하는 문제이다.

문제를 푸는 로직은 algorithm 헤더의 std::sort를 이용하는 방법이 가장 쉽다.

그런데 위와 같이 이중벡터로 되어있을 때에는 sort 함수의 세번째 인자에서 오류가 발생한다.

 

(1) std::sort 함수

- sort 함수의 첫번째와 두번째 인자는 범위를 가리킨다. 이 때 반복자 형식으로 지정해주어야 한다.

(vec.begin(), vec.end(), vec.begin + 1 ; 두번째 원소 등등...)

 

- sort 함수의 세번째 인자는 정렬을 어떻게 할 것인지 지정할 수 있다.

(less<int> ; 숫자 기준 오름차순, greater<int> ; 숫자 기준 내림차순, less<char> ; char형 기준 오름차순 등등 ...)

 

- 세번째 인자인 compare는 벡터의 각 원소를 i, 다음 원소를 j로 지정하고 둘을 비교해서 참과 거짓을 리턴한다.

단, 이중벡터를 비교할 때에는 i와 j가 집합으로 주어지기 때문에 오류가 발생한다.

 

- 따라서 이중벡터 안의 벡터를 비교하도록 cmp 함수를 따로 만들어준다.

 

cmp 함수를 이용해서 첫번째 원소(가로)를 기준으로 내림차순으로 정렬하여, 최대 가로 값을 추출했다면, 다음은 두번째 원소(세로)를 기준으로 내림차순으로 정렬하여 최대 세로 값을 추출해야 한다. 이것은 cmp2 함수로 구현되어 있다.

따라서 주어진 이중벡터를 세로를 기준으로 정렬하여 값을 추출한다.

최종 값은 두 최대값을 곱한 값이다.