백준 1931번. 회의실 배정 (Python / 파이썬)

문제

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

입력

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

출력

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

---

문제 풀이

각 회의에 대해 시작 시간과 끝나는 시간이 주어져 있다고 할 때,

각 회의가 겹치지 않게 하면서 회의실을 사용할 수 있는

회의의 최대 개수를 찾는 문제입니다.

 

처음 이 문제를 해결 할 때

시작 시간이 빠른 순으로 정렬하여 문제를 해결하려 했습니다.

시작 시간이 빠른 순으로 정렬하면

각 회의 마다 회의실을 사용할 수 있는 회의의 개수를

세어준 뒤 그 중 최대 값을 계산해주면 됩니다.

각 회의마다 만약 어떤 회의의 시작 시간이 해당 회의의

끝나는 시간보다 크거나 같다면 회의실을 사용할 수 있는 것이므로

count를 세어주고 그 회의를 다음 비교대상으로 갱신해주었습니다.

하지만 이렇게 해결하면 시간 초과 판정을 받게 됩니다. 

 

파이썬에서 제공하는 함수 대부분은 시간을 많이 차지하지 않는 편입니다.

따라서 알고리즘 자체에 문제가 있다고 파악하고

이중 for문을 이용하는 방법 말고 다른 방법이 있을지 고민해보았습니다.

 

다른 방법은 시작 시간이 아닌 끝나는 시간이 빠른 순으로 정렬하는 것이었습니다.

끝나는 시간이 빠른 순으로 정렬하게 되면

어떤 회의의 끝나는 시간보다 시작 시간이 같거나 큰 회의를 선택해도

 끝나는 시간이 항상 빠르므로 최선의 선택이 됩니다.

따라서 모든 회의를 탐색하지 않아도

첫번째 회의부터 시작 시간이 끝나는 시간보다 더 큰 회의를 선택해가며

한 번만 탐색하면 되므로 시간 초과 판정을 받지 않을 수 있습니다.

---

My Code - 정답 코드

n = int(input()) # 회의의 개수

time = []
for _ in range(n):
    start, end = map(int,input().split()) # 시작 시간, 끝나는 시간
    time.append((start, end)) # list에 삽입

# 끝나는 시간을 우선으로 정렬, 끝나는 시간이 같으면 시작 시간 기준으로 정렬
time.sort(key = lambda x: (x[1], x[0])) 

tmp = 0 
count = 0 # 회의실 사용 가능한 회의의 개수를 셀 변수
for i in time:
    if tmp <= i[0]: # 시작시간이 끝나는 시간보다 크거나 같으면
        count += 1  # 개수를 세어줌
        tmp = i[1]  # 다음 비교 값 갱신

print(count)

My Code - 시간 초과 판정 코드

n = int(input()) # 회의의 개수

time = []
for _ in range(n):
    start, end = map(int,input().split()) # 시작 시간, 끝나는 시간
    time.append((start, end)) # list에 삽입

# 시작 시간을 우선으로 정렬, 시작 시간 같으면 끝나는 시간 기준으로 정렬
time.sort(key = lambda x: (x[0], x[1]))

max_value = 0 # 최대 값을 계산할 변수
for i in time:
    count = 1 # 각 회의마다 가능한 회의의 개수를 셀 변수
    tmp = i[1] # 초기 비교 값
    for j in time:
        if j[0] >= tmp: # 시작시간이 끝나는 시간보다 크거나 같으면
            count += 1  # 개수를 세어줌
            tmp = j[1]  # 다음 비교 값 갱신
    max_value = max(max_value, count) # 세어준 값과 max값을 비교해 최대 값 갱신

print(max_value)