알고리즘 연습/그리디 알고리즘

[🥈1 / 백준 1946 / 파이썬] 신입 사원

김세진 2021. 10. 1. 23:57
반응형

 

 

1946번: 신입 사원

첫째 줄에는 테스트 케이스의 개수 T(1 ≤ T ≤ 20)가 주어진다. 각 테스트 케이스의 첫째 줄에 지원자의 숫자 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개 줄에는 각각의 지원자의 서류심사 성

www.acmicpc.net

 

문제

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

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

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

입력

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

출력

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

 

예제 입력 

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

예제 출력 

4 3














 

풀이

 

다른 모든 지원자와 비교하여 최소 어느 한 가지 성적이 높은 지원자만 카운트하는 문제이다.

주의해야 할 점은 입력으로 주어진 숫자가 점수가 아닌 '순위' 라는 것이다.

최대 20 * 100,000의 연산을 필요로 하므로 O(N^2)의 알고리즘으로는 해결할 수 없다.

 

따라서 그리디 알고리즘으로 해결할 것이다.

우선 지원자의 성적을 첫 번째 성적을 기준으로 오름차순으로 정렬한다.

 

이제 첫 번째 기준으로 성적이 정렬되었으니, 뒤의 지원자는 앞의 지원자보다 무조건 성적 하나가 낮게 된다.

그럼 이제 뒤의 성적만 비교하여, 뒤의 지원자가 앞의 모든 지원자보다 두 번째 성적이 높다면 채용 대상이 된다.

 

채용 대상으로 판별될 때마다 답을 하나씩 카운트 해주며 두 번째 최고 성적도 계속 갱신해 주자.

 

import sys
input = sys.stdin.readline

for _ in range(int(input())):
    n = int(input())
    lst = [tuple(map(int,input().split())) for _ in range(n)]
    lst.sort()
    
    ans = 1
    second_max = lst[0][1]
    for i in range(1,n):
        if lst[i][1] < second_max:
            second_max = lst[i][1]
            ans+=1
    print(ans)

 

지원자를 입력받을 때 tuple 형식으로 이어받는 방법으로 풀었는데, 다른 코드와 비교해보니 최악의 경우 10만개의 튜플로 이루어진 리스트를 20번 가까이 정렬해야 한다는 것이 그다지 효율이 좋지 않은 것 같다.

 

따라서 어차피 왼쪽 성적은 정렬할 것이니 인덱스처럼 받을 리스트를 미리 만들어두고 리스트에 두 번째 성적만 담아 처리했다.

 

import sys
input = sys.stdin.readline

for _ in range(int(input())):
    n = int(input())
    lst = [0]*(n+1)
    for _ in range(n):
        a,b = map(int,input().split())
        lst[a]=b
        
    ans = 1
    second_max = lst[1]
    for i in range(2,n+1):
        if lst[i] < second_max:
            second_max = lst[i]
            ans+=1
    print(ans)

 

반응형