파이썬 그리디 알고리즘 백준 1931 회의실 배정

문제

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

입력

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

출력

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

풀이

가장 많은 회의를 해야 하므로 일찍 시작해서 일찍 끝나는 순으로 회의를 진행하는 것이 좋다.

입력받은 회의의 시작시간과 끝나는시간에 대한 정보를 리스트에 저장한후

sorted함수를 이용하여 시작시간으로 정렬, 끝나는 시간으로 정렬한다.

처음 시작가능한 시간 startTime은 0으로 초기화하고

이후 반복문을 통해 리스트를 순회하면서 시작시간 startTime 보다 리스트의 시작시간이 늦거나 같으면

count를 증가시키고 startTime을 리스트의 끝나는 시간으로 변경한다.

이를 통하여 시작시간이 일찍인 회의중 끝나는 시간이 빠른 회의부터 시작해 가능한 회의 수를 셀 수 있다.

 

def count(meeting):
    mCount = 0
    startTime = 0
    for time in meeting:
        if(time[0]>=startTime):
            startTime = time[1]
            mCount = mCount+1
    return mCount


if __name__ == "__main__":
    N = int(input())
    meeting = [ list(map(int,input().split())) for _ in range(0,N) ]
    meeting = sorted(meeting, key=lambda x: x[0])
    meeting = sorted(meeting, key=lambda x: x[1])
    print(count(meeting))