[1931] 회의실 배정

[Silver I] 회의실 배정 - 1931

문제 링크

성능 요약

메모리: 89072 KB, 시간: 1056 ms

분류

그리디 알고리즘, 정렬

문제 설명

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

입력

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

출력

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

아이디어

처음에는 가능한 모든 경우를 구하면서, 가장 많이 선택할 수 있는 회의의 수를 갱신시키는 방식으로 구현했었다.
하지만, 역시나 시간 초과가 나버렸고, 다시 고민해보다 해결법이 떠오르지 않아 답을 찾아보았다.

출처

결국 핵심은 주어진 시간 동안 최대한 많은 회의를 구성하려면,
‘서로 겹치지 않은 활동에 대하여 종료시간이 더 빠르면 더 많은 회의를 선택할 수 있다.’ 는 점이었다.

따라서, 해야할 것은

1. ‘회의가 끝나는 시간’을 기준으로 정렬(끝나는 시간이 같은 경우에는 시작 시간이 빠른 회의가 앞에 오도록 구성 -> 시작과 끝시간이 같은 회의가 존재하므로..)
2. 다음 회의의 시작 시간을 고려하여, 겹치지 않도록 선택

여기서 복습하게 된 것은 자바에서 Comparator의 compare 메소드를 구현하는 람다식을 작성하는 법이었다.
순위가 겹치는 상황에서 다른 정렬 기준을 반환하도록 코드를 구성하는 방법을 경험해보았다.

답안코드

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

public class 최준혁_1931 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int n = Integer.parseInt(br.readLine());

        List<List<Integer>> schedule = new ArrayList<>();
        for (int i = 0; i < n; i++) {
            List<Integer> room = Arrays.stream(br.readLine().split(" "))
                    .map(Integer::valueOf)
                    .collect(Collectors.toList());
            schedule.add(room);
        }

        // 끝나는 시간 순으로 정렬
        schedule.sort((i1, i2) -> {
            // 시작과 끝이 동일한 회의 시간도 존재하기 때문에 끝나는 시간이 같은 경우 시작 시간이 빠른 회의가 앞에 오도록 정렬해야 함
            if (i1.get(1) - i2.get(1) == 0)
                return i1.get(0) - i2.get(0);
            return i1.get(1) - i2.get(1);
        });

        int chosen_end_time = 0;
        int count = 0;
        for (int i = 0; i < n; i++) {
            if (chosen_end_time > schedule.get(i).get(0))
                continue;

            chosen_end_time = schedule.get(i).get(1);
            count += 1;
        }

        System.out.println(count);
    }
}

BOJ 카테고리 내 다른 글 보러가기