[정렬] 백준 1931번 회의실 배정 java(자바)

https://www.acmicpc.net/problem/1931

1931번: 회의실 배정

문제

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

입력

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

출력

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

코드

import java.util.*;

public class No1931 {
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int[][] arr = new int[N][2];

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

        Arrays.sort(arr, (o1, o2) -> {
            if (o1[1] == o2[1])
                return o1[0] - o2[0];
            else
                return o1[1] - o2[1];
        }); //2차원 배열 정렬
        //1번째 element의 요소가 같으면 0번째 element를 기준으로 정렬하고
        //같지 않으면 1번째 element를 기준으로 정렬한다는 의미

        Stack<Integer> s = new Stack<>();
        s.add(arr[0][1]);
        for(int i=1; i<N; i++){
            if(arr[i][0]>=s.peek()){
                s.add(arr[i][1]);
            }
        } //시작시간이 종료시간보다 빠르거나 같으면 add
        System.out.println(s.size());
    }
}

풀이

종료시간이 빠른것을 기준으로 선택하면 가장 많은 회의를 배정할 수 있다.

1. 시작시간과 종료시간을 2차원 배열로 입력 받고 종료 시간을 기준으로 정렬한다.

이때, 종료 시간이 같지만 시작시간이 다른 경우 시작시간을 기준으로 정렬한다.

2. 스택에 들어있는 종료시간과 다음 회의의 시작시간을 비교하여 시작시간이 종료시간보다 같거나 크면 해당 회의의 종료시간을 스택에 add한다

3. 스택의 사이즈를 출력한다.

 

주의사항

처음에 문제를 풀때 종료시간만 정렬을 해서 정답률이 88%를 넘지 못했다.

정답률이 88%를 못넘어 간다면

아래의 반례를 입력해보고 출력값과 다른 값이 나온다면

종료시간이 동일할때 시작시간이 다른 회의들을 시작기준으로도 정렬을 해야한다.

입력값 3 1 3 8 8 4 8 출력값 3

 

2차원 배열 정렬

2차원 배열에 대한 설명은 아래에 정리되어 있다.

https://seaotter.tistory.com/4

[그리디] 백준 11000번 강의실 배정