[프로그래머스 c++] 2018 서머코딩 > 숫자 게임

문제 설명

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	Result
[5,1,3,7]	[2,2,6,8]	3
[2,2,2,2]	[1,1,1,1]	0

입출력 예 설명

입출력 예 #1

A 팀은 숫자 5를 부여받은 팀원이 첫번째로 출전하고, 이어서 1,3,7을 부여받은 팀원들이 차례대로 출전합니다.
B 팀원들을 4번, 2번, 3번, 1번의 순서대로 출전시킬 경우 팀원들이 부여받은 숫자들은 차례대로 8,2,6,2가 됩니다. 그러면, 첫 번째, 두 번째, 세 번째 경기에서 승리하여 3점을 얻게 되고, 이때가 최대의 승점입니다.

입출력 예 #2
B 팀원들을 어떤 순서로 출전시켜도 B팀의 승점은 0점입니다.

풀이 방법

이 문제의 목적은 B팀이 최다득점을 하기 위한 방법을 구현하는 것입니다.

이를 위해 A, B팀에서 가장 큰 숫자들을 비교해갑니다.

B팀에 있는 가장 큰 숫자가 A팀에 있는 가장 큰 숫자보다 클 경우 이긴 승부라 판단해 answer를 1 높입니다.

그 후, A, B에 있는 가장 큰 숫자들을 가장 큰 숫자 후보에서 제외합니다.

그렇지 않을 경우(B팀에 있는 가장 큰 숫자가 A팀에 있는 가장 큰 숫자보다 작거나 같을 경우), A에 있는 가장 큰 숫자를 가장 큰 숫자 후보에서 제외합니다.

 

A, B vector에서 가장 큰 숫자들을 쉽게 판단하기 위해, A, B를 내림차순으로 정렬 후 비교를 시작합니다.

각 vector에서 첫 번째 인덱스가 가리키는 값들이 최대값이므로, 이를 비교해가며 비교 후 가장 큰 숫자 후보에서 제외하기 위해 

A.erase(A.begin()), B.erase(B.begin())을 사용했었습니다.

하지만 이 경우 정확성 테스트는 모두 통과하나 효율성 테스트가 0점이 나오게 됩니다.

빈번한 함수 호출이 시간을 많이 소모한다고 판단하여, vector에서 값을 삭제하지 않고 각 벡터의 특정 인덱스를 나타내는 변수를 생성하여 문제를 해결했습니다.

아래 풀이는 정확성, 효율성 테스트를 모두 통과합니다.

 

풀이

#include <string>
#include <vector>
#include <algorithm>
#include <functional>

using namespace std;

int solution(vector<int> A, vector<int> B) {
    int answer = 0;
    int team_size = A.size();
    int A_pointer = 0;
    int B_pointer = 0;
    
    //내림차순 정렬
    sort(A.begin(), A.end(), greater<int> ());
    sort(B.begin(), B.end(), greater<int> ());
   
    // B팀이 이길 경우의 수가 0일 경우
    if (A[team_size - 1] >= B[0])
        return 0;
    
    for (int i = 0; i < team_size; i++) {
        if (A[A_pointer] < B[B_pointer]) { // B팀원이 이길 경우
            B_pointer++;
            answer++;
        }
        
        A_pointer++;
    }
    
    return answer;
}