[백준 골드5] 1931번: 회의실 배정 (그리디) - Java

1931번: 회의실 배정 

https://www.acmicpc.net/problem/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

---

풀이 

시도 

1. 재귀 완전 탐색하면 시간 초과가 날 것 같아서 그리디로 접근했는데, 뭔가 잘 풀리지 않음 (끝나는 시간을 currentTime에 저장하면서 계산 -> 무조건 0에서 시작하는 회의를 포함하는... 이상한 풀이)

2. 어쩔 수 없이 재귀로 풀어봄 -> 정답은 나오지만 역시나 시간초과

3. 그리디 다시 접근 -> 시작하는 시간 기준이 아니라 끝나는 시간 기준으로 접근해야 함을 알게 됨!

 

헷갈렸던 논리적 상황 

무조건 빨리 끝나는 회의를 선택하는 것이 이득

- 1 3, 1 4, 4 5 가 있을 때 => 1 3, 1 4 상관없음

- 1 3, 1 4, 3 4 가 있을 때 => 1 3 선택해야

앞에 어느 회의도 없다면, 같은 때에 끝나는 회의 중 무엇을 선택해도 문제 x

- 1 3, 2 3 상관없음 (1 2가 있었다면 1 2가 위의 '무조건 빨리 끝나는 회의를 선택하는 것이 이득'에 의해 선택되었을 것임)

 

=> 정렬을 사용해서 빨리 끝나는 것부터 고른다.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.*;

public class Main{
    public void solution() throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        
        int n = Integer.parseInt(br.readLine());
        int[][] meetings = new int[n][2];
        for(int i=0; i<n; i++){
            StringTokenizer st = new StringTokenizer(br.readLine());
            meetings[i][0] = Integer.parseInt(st.nextToken());
            meetings[i][1] = Integer.parseInt(st.nextToken());
        }
        // 끝나는 시간이 작은 것부터 오름차순 정렬
        Arrays.sort(meetings, (a, b) -> {
            if (a[1] != b[1]) return Integer.compare(a[1], b[1]);
            return Integer.compare(a[0], b[0]);
        });
        
        int count = 0, prevEnd = 0;
        for (int i=0; i<meetings.length; i++) {
            if (meetings[i][0] >= prevEnd) {
                prevEnd = meetings[i][1];
                count++;
            }
        }
        System.out.println(count);
    }
	public static void main(String[] args) throws Exception {
		new Main().solution();
	}
}

 

Arrays.sort(), Comparator에 대하여

Arrays.sort(T[] a): 배열 오름차순 정렬

Arrays.sort(T[] a, Comparator<? super T> c): Comparator 활용해 사용자 정의 정렬 기준 지정.

- 내림차순 정렬(Collections.reverserOrder()), 다차원 배열 정렬, 복합 조건 정렬 등이 가능하다.

 

Comparator 정의 예시

Comparator<T> cmp = (a, b) -> { return 비교결과; };

람다식을 사용해 간결하게 Comparator를 정의할 수 있고, 정렬 기준을 동적으로 지정할 수 있다.

 

위 풀이에서는 2차원 배열에서 i행의 1번 열부터 비교하고 같은 경우 0번 열을 비교하도록 했다.