[백준/BOJ][Python] 1931번 회의실 배정

https://www.acmicpc.net/problem/1931

1931번: 회의실 배정

 

문제

한 개의 회의실이 있는데 이를 사용하고자 하는 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

 

아이디어

처음에 생각했던 풀이 방식은 시작 시간을 x, 종료 시간을 y라고 할 때, y가 가장 작은 값을 구하고 그 다음에는 y보다 큰 x 중 x의 짝인 y가 가장 작은 값을 찾음으로써 하나씩 구해야된다고 생각했다.

# 시간 초과
N = int(input())
x, y = [], []
result = 0
temp = 0

for i in range(N):
    a, b = map(int, input().split())
    x.append(a)
    y.append(b)

for n in range(N):
    if result == 0:
        for j in range(N):
            if y[j] == min(y):
                result += 1
                temp = y[j]
                break
    else:           
        for j in range(N):
            status = 0
            status_2 = 0
            for k in range(N):
                if x[k] >= temp:
                    status_2 = 1
                    if status == 0:
                        index = y[k]  
                        status = 1
                    else:
                        if y[k] < index:
                            index = y[k] 
            if status_2 == 1:
                for i in range(N):
                    if y[i] == index:
                        result += 1
                        temp = y[i]
                        
                        break


print(result)

누가 봐도 반복문이 많아서 시간초과가 났다. 이걸 어떻게 해야 될지 몰라서 고민을 엄청 하다가 결국 검색해서 찾아봤다.

 

우선 처음 회의 시간을 입력할 때, 리스트에 리스트로 시작, 종료 시간을 입력해준 것을 append 해준다. 그 다음 시작 시간을 기준으로 정렬해준다.meeting = sorted(meeting, key = lambda x : x[0] 코드를 사용했는데 key, lambda를 사용해 정렬해주는 것은 처음 배웠다. x[0]은 시작 시간을 기준으로 정렬해준다는 뜻이고 종료 시간 기준으로 정렬하려면 x[0]을 x[1]로 바꿔주면 된다.

코드

n = int(input())
meeting = []
for i in range(n):
    meeting.append(list(map(int, input().split())))

meeting = sorted(meeting, key = lambda x : x[0]) # 시작 시간 기준 정렬
meeting = sorted(meeting, key = lambda x: x[1]) # 종료 시간 기준 정렬

result = 1
end_time = meeting[0][1] # 기준 종료 시간

for i in range(1, n): # 1부터 하는 이유는 0번째 리스트가 이미 첫 회의이기 때문(그래서 result = 1로 시작)
    if end_time <= meeting[i][0]: # 시작 시간이 기준 종료 시간보다 크거나 같아야함
        end_time = meeting[i][1] # 기준 종료 시간 재설정
        result += 1

print(result)