[Java][백준][1931번] 회의실 배정

문제 설명 (링크)

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

 

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

 

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

 

생각

그리디 알고리즘의 문제이다. 그리디 알고리즘의 경우 내가 어떤 부분을 우선으로 설정할 것인지 잘 정해주어야 한다. 회의실 배정 문제의 경우 내가 우선할 수 있는 것은 3가지가 있다. 회의의 시작시간, 회의의 종료시간, 회의하는 시간이다. 위 문제에서 바라는 것은 최대 사용할 수 있는 회의 수를 출력하는 것이므로 회의의 종료시간에 우선을 맞춰야 한다. 회의가 가장 먼저 끝나는 것부터 먼저 한다면 최대한 많은 수의 회의를 할 수 있을 것이다.

 

따라서 회의 시간이 주어지면 회의가 끝나는 시간을 기준으로 오름차순으로 정렬해야 한다. 후에 회의가 끝나는 시간에 맞춰 다음 회의의 시작 시간이 가장 가까운 회의부터 쭉 실행한다면 답이 나올 것이다.

 

위 문제에 경우 푸는 방법은 시작시간과 종료시간을 갖는 회의 노드를 만들어 풀거나, 2차원 배열을 만들어 풀면 될 것이다. 물론 나는 친절하게 두 방법 모두 코드를 넣을 것이다. 또 여기서 중요한 것은 회의가 끝나느 식나을 기준으로 오름차순을 정리하는 것이다. 이는 Compartor, Comparable 등을 사용하여 직접 설정해 주어야 한다.

 

코드

1. 2차원 배열을 사용하는 방법.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;
import java.util.StringTokenizer;
 
public class Main {
 
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        int[][] time = new int[N][2];
        
        for(int i=0; i<N; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            for(int j=0; j<2; j++) {
                time[i][j] = Integer.parseInt(st.nextToken());
            }
        }
        
        //끝나는 시간을 기준으로 정렬을 하고 만약 같을 경우에는 시작시간을 기준으로 한다.
        Arrays.sort(time, new Comparator<int[]>() {
            @Override
            public int compare(int[] arg0, int[] arg1) {
                if(arg0[1] == arg1[1]) {
                    return arg0[0] - arg1[0];
                } else {
                    return arg0[1] - arg1[1];
                }
            }
        });    
        
        int end = -1;
        int count = 0;
        
        //끝나는 시간을 계속 수정해주며 회의를 진행한다.
        for(int i=0; i<N; i++) {
            if(time[i][0] >= end ) {
                end = time[i][1];
                count++;
            }
        }
        
        System.out.println(count);
    }
 
}

2. time class를 새로 만드는 방법.

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;
 
public class Main2 {
 
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(br.readLine());
        times[] time = new times[n];
        
        for(int i=0; i<n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int b = Integer.parseInt(st.nextToken());
            int c = Integer.parseInt(st.nextToken());
            
            time[i] = new times(b,c);
        }
        
        Arrays.sort(time);
                
        int count = 0;
        int end = -1;
        
        for(int i=0; i<n; i++) {
            if(time[i].start >= end) {
                end = time[i].end;
                count++;
            }
        }
        
        System.out.println(count);
        
    }
 
}
//time 클래스를 새로 만든다.
class times implements Comparable<times>{
    int start;
    int end;
    
    public times(int start, int end) {
        this.start = start;
        this.end = end;
    }
 
    @Override
    public int compareTo(times arg0) {
        if(this.end == arg0.end) {
            return Integer.compare(this.start, arg0.end);
        } else {
            return Integer.compare(this.end, arg0.end);
        }
    }
}