[1931 - JAVA] 회의실 배정

문제

 

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

입력

 

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

출력

 

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

 

첫 번째 풀이

 

일단 오답이다 정렬 부분에서 종료시각이 동일한 경우를 고려하지 않고 정렬을 했다. 때문에 마지막 값이 동일하고 시작 값이 다른 경우 위치를 바꿔야 하는데 해당 코드로는 불가능 하다.

import java.util.Arrays;
import java.util.Scanner;
import java.util.Comparator;

public class Main {

    public static void main(String[] args) {

        Scanner in = new Scanner(System.in);

        int N = in.nextInt();

        int[][] time = new int[N][2];

        for (int i = 0; i < N; i++) {
            time[i][0] = in.nextInt();    // 시작시간
            time[i][1] = in.nextInt();    // 종료시간
        }

        Arrays.sort(time, Comparator.comparingInt(o -> o[1]));

        for (int[] t : time) {
            System.out.println(t[0] + " " + t[1]);
        }
        int count = 0;
        int prev_end_time = 0;

        for (int i = 0; i < N; i++) {
            if (prev_end_time <= time[i][0]) {
                prev_end_time = time[i][1];
                count++;
            }
        }

        System.out.println(count);
    }
}

 

수정 코드

 

람다 식을 이용해 조건을 추가해보자. 

public class Main {

    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);

        int n = sc.nextInt();

        int[][] meetings = new int[n][2];

        for (int i = 0; i < n; i++) {
            meetings[i][0] = sc.nextInt();
            meetings[i][1] = sc.nextInt();
        }

        // 끝나는 시간을 기준으로 정렬
        Arrays.sort(meetings, (p1, p2) -> {
            if (p1[1] == p2[1]) {
                // 비교 값이 음수라면 p1 값보다 앞으로 온다.
                return p1[0] - p2[0];
            }
            return p1[1] - p2[1];
        });

        int count = 0;

        int prevEnd = 0;

        // 각 회의에 대해 순회하면 최대 개수를 찾기
        for (int i = 0; i < n; i++) {
            int start = meetings[i][0];
            int end = meetings[i][1];

            // 이전 회의의 종료 시간보다 시작 시간이 늦다면 회의를 선택
            if (start >= prevEnd) {
                count++;
                prevEnd = end;
            }
        }

        System.out.println(count);
    }
}