[Silver I] 회의실 배정 - 1931

[Silver I] 회의실 배정 - 1931

문제 링크

성능 요약

메모리: 56780 KB, 시간: 312 ms

분류

그리디 알고리즘, 정렬

제출 일자

2024년 10월 30일 14:43:57

문제 설명

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

입력

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

출력

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

 

 

풀이 

import sys

sys.stdin = open("input.txt", "r")

n = int(sys.stdin.readline())
lst = []
for i in range(n):
    start, end = map(int, sys.stdin.readline().split())
    tmp = []
    tmp.append(start)
    tmp.append(end)
    lst.append(tmp)

count = 0

## 끝나는 시간이 같으면 시작 시간이 더 작은 순으로 정렬 
sorted_l = sorted(lst, key=(lambda x: (x[1], x[0])))


if(len(sorted_l) != 0):
    
    start_f , end_f = sorted_l[0][0] , sorted_l[0][1]

    count += 1

    for i in sorted_l[1:]:
        start_n = i[0]
        end_n = i[1]

        if(end_f <= start_n):
            count += 1
            end_f = end_n


print(count)

 

 

그리디 접근은 맞게 풀었지만 이상하게 답이 틀렸다고 했는데, 그 이유는 정렬 때문이었다.

테스트 케이스에는 없지만, 단순히 끝나는 시간으로 오름차순 정렬만 한 경우 뿐만 아니라 끝나는 시간이 같은 시간 있으면 시작 시간이 더 작은 것으로 다시 다시 정렬하는 경우를 포함해서 생각해주어야한다. 

sorted_l = sorted(lst, key=(lambda x: (x[1], x[0])))

이 람다함수는 (x[1], x[0])로 튜플을 반환하여, 첫 번째로 x[1]을 기준으로, 두 번째로 x[0]을 기준으로 정렬한다.