Greedy 백준 알고리즘 자바 1931 ‘회의실 배정’ 문제풀이

Problem

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

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

Output
첫째 줄에 최대 사용할 수 있는 회의 수를 출력하여라.

Solution

대체로 스케줄링 문제는 그리디로 풀면된다고 한다. 뭐가 딱히 그리디인지 모르겠으나 그냥 봤을때, 끝나는 시간이 가장 빠른 것을 선택해야 최대한 많은 회의를 진행할 수 있을 것 같아 우선 끝나는 시간 순으로 정렬을 한다. 이때 배열이 2차원이므로, comparator를 사용해 종료 값이 같으면 시작 시간이 가장 짧은 애로 정렬하도록 했다.

Code

import java.util.Arrays;
import java.util.Scanner;
public class Main {
	public static void main(String[] args) {
		Scanner scan = new Scanner(System.in);
		int n = scan.nextInt();
		int[][] times = new int[n][2]; // 0 : start 1 : finish
		
		int min = -1;
		for(int i = 0; i < n; i++) {
			times[i][0] = scan.nextInt();
			times[i][1] = scan.nextInt();
		}
		
		Arrays.sort(times, (a,b) -> a[1] == b[1]? a[0] - b[0] : a[1] - b[1]);
		int cnt = 0; 
		for(int i = 0; i < n; i++) {
			if(times[i][0] >= min) {
				min = times[i][1];
				cnt++;
			}
		}
		System.out.println(cnt);
	}
}