[Python] 백준 1946 신입 사원 (Greedy)

📌 문제

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

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

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

입력

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

출력

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

예제 입력 1

2
5
3 2
1 4
4 1
2 3
5 5
7
3 6
7 3
4 2
1 4
5 7
2 5
6 1

예제 출력 1

4
3

📌 풀이

💬 Code

import sys
input = sys.stdin.readline

for _ in range(int(input())):
    n = int(input())
    people = [list(map(int, input().split())) for _ in range(n)]
    people.sort()

    cnt = 0
    minimum = n+1
    for person in people:
        if person[1] < minimum:
            cnt += 1
            minimum = person[1]
    print(cnt)

💡 Solution

탈락 조건 : 어떤 지원자 A의 성적이 다른 어떤 지원자 B의 성적에 비해 서류 심사 결과와 면접 성적이 모두 떨어진다면 A는 결코 선발되지 않는다.

예제의 두 번째 테스트케이스를 보자. 입력을 받고 서류 성적을 기준으로 정렬까지 하면 people은 [[1, 4], [2, 5], [3, 6], [4, 2], [5, 7], [6, 1], [7, 3]]이 된다.

[1, 4]는 서류등수가 1등이라 탈락을 피한다.

[2, 5]는 서류등수순으로 정렬했기 때문에 서류등수는 당연히 앞선 지원자들보다 낮다. 따라서 면접등수만 앞선 지원자들보다 높으면 탈락을 피한다. 앞선 지원자들의 면접등수 중 가장 높은 등수를 구해서 비교해주면 되는데, 아쉽지만 앞선 지원자는 4등, 이 지원자는 5등이므로 탈락이다.

[3, 6]도 마찬가지로 서류등수가 앞선 지원자들보다 낮다. 앞선 지원자들의 면접등수 중 가장 높은 등수가 4등인데, 이 지원자는 6등이므로 탈락이다.

[4, 2]도 마찬가지로 서류등수가 앞선 지원자들보다 낮다. 앞선 지원자들의 면접등수 중 가장 높은 등수가 4등인데, 이 지원자는 2등이므로 탈락을 피한다.

아이디어는 위와 같다! 그리고 이 아이디어를 코드로 작성하면 아래와 같다.

cnt = 0
minimum = n+1
for person in people:
    if person[1] < minimum:
        cnt += 1
        minimum = person[1]
    print(cnt)