[백준 1946번] 신입사원(파이썬)

문제

언제나 최고만을 지향하는 굴지의 대기업 진영 주식회사가 신규 사원 채용을 실시한다. 인재 선발 시험은 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

 

---

 

풀이

이 문제는 입력에 점수가 주어진게 아니라 순위가 주어졌습니다.
선발 원칙은 어떤 지원자가 다른 지원자에 비해 1차와 2차 성적이 모두 떨어진다면 선발되지 않습니다.
이런 원칙을 만족하면서 선발할 수 있는 신입사원의 최대 인원수를 구해야 합니다.

예제 1
                       1차     2차
1번 지원자 :    3등     2등
2번 지원자 :    1등     4등
3번 지원자 :    4등     1등
4번 지원자 :    2등     3등
5번 지원자 :    5등     5등

1차를 기준으로 오름차순으로 정렬하면
(1, 4) - (2, 3) - (3, 2) - (4, 1) - (5, 5)
1번, 2번, 3번, 4번 지원자는 2차는 다른 지원자들에 비해 성적이 좋고, 5번 지원자는 둘다 성적이 모두 떨어집니다.
결국 선발된 인원은 4명으로 정답은 4가 되어 출력이 됩니다.

코드를 순서대로 작성한다면
1. 테스트 케이스 수 입력
2. 각 테스트 케이스의 지원자 수 입력 후 각 지원자들의 성적 입력
3. 입력된 성적들을 정렬하기 위해 리스트 선언 후 리스트에 저장
4. 1차 점수가 가장 좋은 지원자의 2차 성적을 변수에 선언하고 카운트 1부터 저장
5. 1차 점수가 두번 째로 좋은 지원자부터 2차 점수를 비교해 등수가 높으면 변수에 저장하고 카운트 +1

※ sys.stdin.readline 을 쓰는 이유는 입력 시간 초과 때문입니다.
   한 두개의 입력일 때에는 input을 사용하고, 문제처럼 입력이 많을 때 sys.stdin.readline을 사용합니다.

---

작성 코드

import sys
input = sys.stdin.readline # 입력이 많을 때 시간초과 현상 방지

N = int(input()) # 테스트 케이스 수 입력
for _ in range(N):
    A = int(input()) # 케이스 수만큼 지원자 수 입력
    
    data = [] # 지원자 성적을 저장할 리스트 선언
    for _ in range(A): # 지원자 수만큼 지원자들의 성적 입력
        a, b = map(int, input().split())
        data.append((a, b))
    data.sort() # 지원자 성적을 오름차순으로 정렬
    
    answer = 1
    min = data[0][1] # 1차 성적이 가장 좋은 지원자의 2차 성적을 선언 후 카운트 1부터 시작
    
    for i in range(1, A): # 1차 성적이 두번째로 좋은 지원자부터 2차 성적을 비교
        if data[i][1] < min: # 성적이 좋으면 min에 저장 후 카운트 +1
            min = data[i][1]
            answer += 1
    print(answer)