백준 1931번

회의실 배정

문제

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

입력

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

출력

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

예제

알고리즘 분류

• 그리다 알고리즘
• 정렬

코드

import java.util.*;
import java.io.*;

public class Main {	
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		
		int N = Integer.parseInt(br.readLine());
		
		int arr[][] = new int[N][2];
		
		StringTokenizer st;
		for(int i=0; i<N; i++) {
			st = new StringTokenizer(br.readLine(), " ");
			arr[i][0] = Integer.parseInt(st.nextToken());
			arr[i][1] = Integer.parseInt(st.nextToken());
		}
		
		Arrays.sort(arr, new Comparator<int[]>() {
			@Override
			public int compare(int a[], int b[]) {
				if(a[1] == b[1])
					return a[0] - b[0];
				
				return a[1] - b[1];
			}
		});
		
		int count = 0;
		int time = 0;
		
		for(int i=0; i<N; i++) {
			if(time <= arr[i][0]) {
				count++;
				time = arr[i][1];
			}
		}
		
		System.out.println(count);
	}
}

풀이

2차원 배열을 정렬하기 위해 Comparator 인터페이스를 활용하여 sort함수를 재정의하는 것이 핵심이다.

위 문제에서는 회의가 끝나는 시간을 기준으로 정렬하였는데, 그 이유는
회의가 시작하는 시간을 기준으로 하면 끝나는 시간을 뒤에 알수 없기 때문에 추가적인 반복문이 필요해 시간이 오래 소요된다. 그리고 먼저 시작한다고 해서 그것이 가장 좋은 경우의 수라는 보장도 없다.

sort 재정의 부분 내용 중, if 부분은 회의가 끝나는 시간이 같을 때 회의 시작 시간을 비교하는 내용이다.