[Python] 백준 / 회의실 배정 / 1931번 / 그리디

그리디 개념 바로가기

문제

회의실 배정 문제 링크

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

입력

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

11
1 4
3 5
0 6
5 7
3 8
5 9
6 10
8 11
8 12
2 13
12 14

출력

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

4

접근 방식

1. 끝나는 시간과 시작 시간 순으로 오름차순 정렬한다

2. 이전 회의의 끝나는 시간보다 크거나 같은 시작 시간을 정렬된 순서에 따라 반복적으로 찾는다

코드

import sys

n = int(input())
num_list = []

# 2차원 리스트로 입력받기
for _ in range(n):
    start, end = map(int,sys.stdin.readline().split())
    num_list.append([start,end])

# 끝나는 시간으로 오름차순 정렬 후 끝나는 시간이 같으면 시작 시간을 오름차순 정렬
num_list = sorted(num_list, key=lambda x : (x[1],x[0]))

# 첫 미팅
now_meeting =  num_list[0]
cnt = 1
idx = 1

# 이전 회의의 끝나는 시간보다 크거나 같은 시작 시간을 찾으면 현재 회의를 바꾸고 카운팅
while idx < len(num_list):
    if now_meeting[1] <= num_list[idx][0]:
        now_meeting = num_list[idx]
        cnt+=1
    
    idx +=  1

print(cnt)