프로그래머스[Lv.3] - 숫자 게임

 

 

문제 설명

xx 회사의 2xN명의 사원들은 N명씩 두 팀으로 나눠 숫자 게임을 하려고 합니다. 두 개의 팀을 각각 A팀과 B팀이라고 하겠습니다. 숫자 게임의 규칙은 다음과 같습니다.

• 먼저 모든 사원이 무작위로 자연수를 하나씩 부여받습니다.
• 각 사원은 딱 한 번씩 경기를 합니다.
• 각 경기당 A팀에서 한 사원이, B팀에서 한 사원이 나와 서로의 수를 공개합니다. 그때 숫자가 큰 쪽이 승리하게 되고, 승리한 사원이 속한 팀은 승점을 1점 얻게 됩니다.
• 만약 숫자가 같다면 누구도 승점을 얻지 않습니다.

전체 사원들은 우선 무작위로 자연수를 하나씩 부여받았습니다. 그다음 A팀은 빠르게 출전순서를 정했고 자신들의 출전 순서를 B팀에게 공개해버렸습니다. B팀은 그것을 보고 자신들의 최종 승점을 가장 높이는 방법으로 팀원들의 출전 순서를 정했습니다. 이때의 B팀이 얻는 승점을 구해주세요.
A 팀원들이 부여받은 수가 출전 순서대로 나열되어있는 배열 A와 i번째 원소가 B팀의 i번 팀원이 부여받은 수를 의미하는 배열 B가 주어질 때, B 팀원들이 얻을 수 있는 최대 승점을 return 하도록 solution 함수를 완성해주세요.

---

 

 

 

 

 

 

제한 사항

• A와 B의 길이는 같습니다.
• A와 B의 길이는 1 이상 100,000 이하입니다.
• A와 B의 각 원소는 1 이상 1,000,000,000 이하의 자연수입니다.

---

 

 

 

 

 

 

풀이

문제를 요약하면, A와 B의 요소를 하나씩 매치하여 B의 요소가 더 큰 경우의 수를 최대한으로 늘려야 한다.

모든 요소를 하나씩 비교하며 최댓값을 구해도 되지만, 제한 사항을 보면 각 배열의 최대 개수가 100,000개이기 때문에 $100,000^2$의 시간 복잡도가 요구되기 때문에 시간초과가 발생할 것이다.

 

그렇다면 다른 접근법이 필요하다.

바로 정렬을 이용하는 방법이다.

A와 B의 모든 요소를 하나의 배열에 넣은 뒤 정렬을 수행한다.

이후, 앞에서부터 하나씩 조회하며 B의 요소가 나온다면 이전에 나온 A의 요소가 있는지 확인하면 된다.

만약, 이전에 나온 A의 요소가 있다면 B가 이길 수 있는 경우이다.

그렇지 않다면, 해당 B의 요소는 A를 이길 수 없다는 것이기 때문에 그냥 넘어가면 된다.

 

예를 들어, A가 [5, 1, 3 ,7], B가 [2, 2, 6, 8] 일 때, 정렬을 하면 다음과 같이 된다.

1	2	2	3	5	6	7	8
A	B	B	A	A	B	A	B

A의 요소인 1이 가장 먼저 나왔기 때문에 A의 수를 하나 늘린다.

B의 요소인 2가 나왔을 때, 이전에 나온 A의 요소(1)가 있기 때문에 2가 이길 수 있다.

이후, B의 요소인 2가 다시 나왔을 때, 이전에 나온 A의 요소는 이미 이전의 B의 요소가 이겼기 때문에 현재 요소가 이길 수 있는 A의 요소는 없다.

 

여기서 한 가지 주의할 점이 있다.

A와 B의 요소가 같은 경우에는 어느 누구도 승점을 얻을 수 없다.

따라서, 정렬하는 과정에서 A, B 모두 같은 숫자라면 B의 요소가 A의 요소보다 먼저 나오게 해야 한다.

bool cmp(pair<int,int>& a, pair<int,int>& b)
{
    if(a.first == b.first)
    {
        return a.second > b.second;
    }

    return a.first < b.first;
}

---

 

 

 

 

 

전체 코드

#include <string>
#include <vector>
#include <algorithm>
#include <iostream>

using namespace std;

enum
{
    E_A,
    E_B
};


bool cmp(pair<int,int>& a, pair<int,int>& b)
{
    if(a.first == b.first)
    {
        return a.second > b.second;
    }

    return a.first < b.first;
}

int solution(vector<int> A, vector<int> B) {
    int answer = 0;
    
    vector<pair<int, int>> total;
    
    for(int i = 0 ; i < A.size(); i++)
    {
        total.push_back({A[i], E_A});
        total.push_back({B[i], E_B});
    }
    
    sort(total.begin(), total.end(), cmp);
    
    int Acnt = 0;
    
    for(auto t : total)
    {
        if(t.second == E_B)
        {
            if(Acnt > 0)
            {
                answer++;
                Acnt--;
            }
        }
        else Acnt++;
    }
    
    return answer;
}

---