[Silver I] 회의실 배정 - 1931

문제 링크

성능 요약

메모리: 58432 KB, 시간: 300 ms

분류

그리디 알고리즘(greedy), 정렬(sorting)

문제 설명

한 개의 회의실이 있는데 이를 사용하고자 하는 N개의 회의에 대하여 회의실 사용표를 만들려고 한다. 각 회의 I에 대해 시작시간과 끝나는 시간이 주어져 있고, 각 회의가 겹치지 않게 하면서 회의실을 사용할 수 있는 회의의 최대 개수를 찾아보자. 단, 회의는 한번 시작하면 중간에 중단될 수 없으며 한 회의가 끝나는 것과 동시에 다음 회의가 시작될 수 있다. 회의의 시작시간과 끝나는 시간이 같을 수도 있다. 이 경우에는 시작하자마자 끝나는 것으로 생각하면 된다.

입력

첫째 줄에 회의의 수 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N+1 줄까지 각 회의의 정보가 주어지는데 이것은 공백을 사이에 두고 회의의 시작시간과 끝나는 시간이 주어진다. 시작 시간과 끝나는 시간은 231-1보다 작거나 같은 자연수 또는 0이다.

출력

첫째 줄에 최대 사용할 수 있는 회의의 최대 개수를 출력한다.

접근

그리디에 입문하기에 좋은 문제이다.

먼저 회의 시간을 2차원 배열에 저장한 후 끝나는 시간을 기준으로 정렬한다. 파이썬으로 여러개의 인덱스를 기준으로 정렬 하는 방법은 해당 링크를 참조하길 바란다.
파이썬 정렬

그리디의 특징인 local optimum을 구하기 위해 끝나는 시간이 제일 빠른 회의를 선택한다.

나는 정렬된 배열에서 첫번째 회의로 변수들을 초기화했기 때문에 첫 회의를 빼고 순회하였다.

순회를 하며 현재 미팅의 시작시간이 전 미팅의 끝나는 시간보다 크거나 같음을 조건으로 걸어 최대한 많은 수의 회의를 카운트 한다.

이 때 start, finish변수를 업데이트 해준다.

코드

#https://www.acmicpc.net/problem/1931
import sys
input = sys.stdin.readline
n = int(input())
meetingTable = []
for _ in range(n):
    meetingTable.append(list(map(int,input().split())))
meetingTable.sort(key = lambda x: (x[1], x[0]))

ans = 1
start,finish = meetingTable[0][0],meetingTable[0][1]
#pick meetings with smallest finish time
for meeting in meetingTable[1:]:
    #start of current must be larger than finish of previous
    if meeting[0] >= finish:
        start = meeting[0]
        finish = meeting[1]
        ans += 1

print(ans)