[백준 1931] 회의실 배정(Python, Java)

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

1931번: 회의실 배정

문제

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

입력

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

출력

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

풀이 설명

1. 처음 풀었을 때 반복문을 2중으로 돌렸더니 시간초과가 났다.

2. 입력받은 meeting배열을 시작시간이 아닌, 종료시간으로 정렬해서 풀어야한다.

3. 무조건 일찍 끝나는 회의들을 최대한 많이 선택한 것이 답이다.

4. 회의 개수를 저장할 변수 cnt, 최근 선택한 회의의 종료시간을 저장할 변수 last를 선언하고, 

   last는 가장 빨리 끝나는 회의인 meeting[0]의 종료시간으로 초기화한다.

5. meeting 배열을 종료시간 기준으로 오름차순 정렬 후, 탐색하며 겹치지 않는 회의를 찾는다.

6. 찾으면 cnt 1 증가, last를 찾은 회의의 종료시간으로 갱신.

Python3 코드

n = int(input())
meeting = []
for i in range(n):
    meeting.append(list(map(int, input().split())))
    
meeting.sort(key = lambda x : (x[1], x[0]))
    
cnt = 0
last = -1
for i in range(n):
    if meeting[i][0] >= last:
        cnt += 1
        last = meeting[i][1]
        
print(cnt)

Java 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;
import java.util.StringTokenizer;

public class Main {

	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		int[][] meeting = new int[N][2];
		for(int i=0; i<N; i++) {
			StringTokenizer st = new StringTokenizer(br.readLine());
			meeting[i][0] = Integer.parseInt(st.nextToken());
			meeting[i][1] = Integer.parseInt(st.nextToken());
		}
		
		Arrays.sort(meeting, new Comparator<int[]>() { //종료시간으로 정렬
			@Override
			public int compare(int[] o1, int[] o2) {
                if(o1[1] - o2[1] == 0)
					return o1[0] - o2[0];
				return o1[1] - o2[1];
			}
		});
		
		int cnt = 0;
		int last = -1;
		for(int i=0; i<N; i++) {
			if(meeting[i][0] >= last) {
				cnt++;
				last = meeting[i][1];
			}
		}
		
		System.out.println(cnt);
	}
}