백준 1931_회의실 배정 / Python

문제

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

입력

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

출력

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

예제 입력 1 

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

예제 출력 1 

4

---

그리디 알고리즘이라는 것만 생각하고 시작시간으로 정렬해서 하나씩 선택하면 될거라 생각해서 풀었다.

## 1931 회의실 배정
#시작 기준
n=int(input())
meeting=[]
for _ in range(n):
  meeting.append(list(map(int,input().split(' '))))

meeting=sorted(meeting, key=lambda x: (x[0],x[1]))	#시작값정렬(같으면 끝값정렬)

fst=meeting[0][1]	#처음 값은 시작이 젤 빠른 값의 끝나는 시간
cnt=1
while fst<meeting[-1][1]:
  for meet in meeting:
    if meet[0]>=fst:	#그 끝나는 시간 이후의 가장 첫번째 회의를 배정
      fst=meet[1]
      cnt+=1

print(cnt)

근데 시작시간을 기준으로 하면 안된다..!!! 

예제에도 나와있듯이 0, 6 이 있지만 첫 회의는 1,4 가 선택되어야함.

 

그래서 끝나는 시간을 기준으로 다시 정렬했다.

## 1931 회의실 배정
#끝 기준
n=int(input())
meeting=[]
for _ in range(n):
  meeting.append(list(map(int,input().split(' '))))

meeting=sorted(meeting, key=lambda x: (x[1],x[0]))	#끝나는 시간 기준으로 정렬

fst=meeting[0][1]
cnt=1
while fst<meeting[-1][1]:
  for meet in meeting:
    if meet[0]>=fst:
      fst=meet[1]
      cnt+=1

print(cnt)

시간초과...나는 시간초과가 제일 싫다...흑흑

 

지금 코드는 끝나는 시간 이후의 첫번째 회의를 찾기위해 모든 회의를 검토한다. 그래서 반복이 너무 많다. 그 반복을 최소화 시켜주었다.

## 1931 회의실 배정
#끝 기준, 반복 최소
n=int(input())
meeting=[]
for _ in range(n):
  meeting.append(list(map(int,input().split(' '))))

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

cnt, cur = 0, 0

for s, e in meeting:
  if s >= cur:	#정렬되어 있으니 순서대로 탐색하면 지금의 end 보다 큰 것 중 첫번째 회의가 자동으로 배정됨.
    cur=e		#current 값 업데이트
    cnt+=1
      
print(cnt)