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

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

1946번: 신입 사원

 

문제

 

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

 

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

 

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

 

입력

 

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

 

출력

 

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

 

풀이

 

지원자들의 서류 성적과 면접 성적을 튜플로 리스트에 저장한 후, 서류 성적을 기준으로 오름차순 정렬한다.

이후 면접 성적이 내림차순이 아닌 지원자들을 탈락처리 하도록 한다.

 

처음에는 서류 성적과 면접 성적을 입력 받을 때 input 함수를 사용하였고, 시간 초과 에러가 발생했다.

다른 곳에서 시간 복잡도를 줄여보고자 하였지만 방법을 찾지 못하였고, 결국 구글링을 통해 여러줄을 입력받을 때는 sys 모듈의 sys.stdin.readline() 함수를 사용하는 것이 더 빠르다는 것을 알 수 있었다.

 

소스 코드

 

import sys

t = int(input())

for i in range(t):
    n = int(input())
    scores = []
    for j in range(n):
        a, b = sys.stdin.readline().split()
        a, b = int(a), int(b)
        scores.append((a, b))
    scores.sort(key = lambda x : x[0])
    
    cnt = 1
    max = scores[0][1]
    for j in range(1, n):
        if max > scores[j][1]:
            cnt += 1
            max = scores[j][1]
    print(cnt)