[백준] 1946 신입 사원 (C++)

링크: https://www.acmicpc.net/problem/1946

1946번: 신입 사원

문제

언제나 최고만을 지향하는 굴지의 대기업 진영 주식회사가 신규 사원 채용을 실시한다. 인재 선발 시험은 1차 서류심사와 2차 면접시험으로 이루어진다. 최고만을 지향한다는 기업의 이념에 따라 그들은 최고의 인재들만을 사원으로 선발하고 싶어 한다.

그래서 진영 주식회사는, 다른 모든 지원자와 비교했을 때 서류심사 성적과 면접시험 성적 중 적어도 하나가 다른 지원자보다 떨어지지 않는 자만 선발한다는 원칙을 세웠다. 즉, 어떤 지원자 A의 성적이 다른 어떤 지원자 B의 성적에 비해 서류 심사 결과와 면접 성적이 모두 떨어진다면 A는 결코 선발되지 않는다.

이러한 조건을 만족시키면서, 진영 주식회사가 이번 신규 사원 채용에서 선발할 수 있는 신입사원의 최대 인원수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에는 테스트 케이스의 개수 T(1 ≤ T ≤ 20)가 주어진다. 각 테스트 케이스의 첫째 줄에 지원자의 숫자 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개 줄에는 각각의 지원자의 서류심사 성적, 면접 성적의 순위가 공백을 사이에 두고 한 줄에 주어진다. 두 성적 순위는 모두 1위부터 N위까지 동석차 없이 결정된다고 가정한다.

출력

각 테스트 케이스에 대해서 진영 주식회사가 선발할 수 있는 신입사원의 최대 인원수를 한 줄에 하나씩 출력한다.

---

풀이

생각보다 까다롭고 생각이 필요한 문제이다.

점수에 두 가지 종류가 있고 두 가지 점수 모두 고려해야 하기 때문이다.

특히 정렬을 어떻게 할 지에 대하여 고민이 많았다.

하지만 점수는 두 종류가 있었고 마땅한 해결책이 더오르지 않아 결국 '서류심사' 순위가 낮은 순으로 정렬을 진행하였다.

(면접시험 순위가 낮은 순, 높은 순 뭐든 상관없다.)

 

서류심사 순위가 낮은 순으로 지원자가 정렬되어 있을 때 푸는 방법은 다음과 같다.

1. 지원자를 뽑는다.

2. 스택이 비거나, 뽑은 지원자보다 면접시험 순위가 높은 지원자가 등장할 때 까지 스택을 pop한다.

(서류심사 순위가 낮은 순으로 정렬되어 있기 때문에 스택 내의 지원자들보다 현재 뽑은 지원자의 서류심사 순위가 높다는 것은 보장되어 있다.)

3. 스택에 뽑은 지원자를 push한다.

4. 정렬된 모든 지원자들에 대하여 1~3을 진행한다.

 

조금 더 알기 쉽게 첫 번째 예제를 보면 다음과 같다.

(5 5)를 스택에 넣는다

 

(4 1)은 (5 5)보다 서류심사, 면접시험 순위가 모두 높기 때문에 (5 5)를 제거하고 (4 1)을 스택에 추가한다.

 

(3 2)는 (4 1)보다 면접시험 순위가 낮기 때문에 스택의 변화 없이 (3 2)를 추가한다.

 

마찬가지로 (2 3)과 (1 4)를 추가하면 위와 같고, 정답은 스택의 사이즈인 4이다.

 

코드

#include <iostream>
#include <queue>
#include <stack>
using namespace std;


int main(){
    ios::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);

    int t;
    cin >> t;

    while (t--){
        int n;
        cin >> n;

        // 서류심사 순위가 낮은 지원자순으로 정렬 
        priority_queue<pair<int, int>> pq;

        for (int i = 0; i < n; i++){
            int a, b;
            cin >> a >> b;

            pq.push({a, b});
        }
        
        stack<pair<int, int>> s;
        // 첫 번째 원소를 집어넣음
        s.push(pq.top());
        pq.pop();

        while (!pq.empty()){
            pair<int, int> target = pq.top();
            pq.pop();
            
            // 면접시험 점수가 더 높은 사람이 나올때까지 스택을 pop
            while (!s.empty() && s.top().second > target.second){
                s.pop();
            }
            s.push(target);
        }

        cout << s.size() << '\n';
    }
}

 

추가로 스택을 pair<int, int>가 아닌 int로 정의하여 면접시험 점수만 저장하더라도 풀 수 있을 것이고

이렇게 진행한다면 메모리를 더 절약할 수 있을 것이다.