백준 1931번 :: 회의실 배정 (Java)

문제 설명

백준 1931번: 회의실 배정 (Silver 2)

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

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

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

문제 풀이

1. 우선순위 큐를 이용하여 회의를 끝나는 시간이 빠른 순으로 저장한다.
   (끝나는 시간이 같다면 시작 시간이 빠른 순으로 저장)
2. 큐에서 회의 정보를 하나씩 꺼내면서 앞의 회의가 끝난 시간과 비교하여 회의실을 사용 가능한 회의의 개수를 센다.
3. 끝나는 시간이 가장 빠른 순으로 처리되므로 2번에서 구한 개수가 최대 개수가 된다.

소스 코드

소스 코드 링크

static class Meeting implements Comparable<Meeting> {
	int start, end;
	public Meeting(int start, int end) {
		this.start = start;
		this.end = end;
	}
	@Override
	public int compareTo(Meeting o) {
		// 끝나는 시간이 같다면 시작 시간이 빠른 회의부터
		if (this.end == o.end) return this.start-o.start;
		// 끝나는 시간이 빠른 회의부터
		else return this.end-o.end;
	}
}

public static void main(String[] args) throws NumberFormatException, IOException {
	BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
	StringTokenizer st = null;
	int N = Integer.parseInt(in.readLine());
	
	// 각 회의의 시작 및 끝나는 시간 입력
	PriorityQueue<Meeting> queue = new PriorityQueue<Meeting>();
	int start, end;
	for (int i = 0; i < N; i++) {
		st = new StringTokenizer(in.readLine(), " ");
		start = Integer.parseInt(st.nextToken());
		end = Integer.parseInt(st.nextToken());
		queue.add(new Meeting(start, end));
	}
	// 회의실을 사용할 수 있는 회의의 최대 개수
	int prev = 0, count = 0;
	for (int i = 0; i < N; i++) {
		Meeting tmp = queue.poll();
		// 회의실을 사용할 수 있다면
		if (prev <= tmp.start) {
			count++;
			prev = tmp.end;
		}
	}
	System.out.println(count);
}

느낀 점

정렬 혹은 우선순위 큐만 잘 이용하면 금방 해결 할 수 있는 문제. 그래서 금방 풀었다. 똑같은 내용을 몇 달 전 수업에서 했었어서 더 금방 푼 것 같지만 그냥 풀었어도 그리 어려운 문제는 아닌 듯!

아무튼 그리디 알고리즘은 욕심쟁이 알고리즘이라는 말처럼 각 단계에서 가장 최적인 답을 선택해서 전체에서도 최적인 답을 구하자는 알고리즘으로, 각 단계에서 최적인 답일지라도 전체에서는 최적이라는 보장이 없으므로 부분의 최적해들의 집합이 전체 문제의 해답이 될 때만 사용할 수 있다.

이번 문제의 경우 앞 회의가 끝나는 시간이 빠를수록 다음 회의가 회의실을 더 빨리 사용할 수 있으므로 회의들을 끝나는 순서가 빠른 순으로 처리를 하면 회의실을 사용할 수 있는 전체 개수가 줄어들 일이 없고, 따라서 그리디 알고리즘을 적용할 수 있다(⌒▽⌒) 다음에도 이렇게 그리디 금방 풀고 싶다... 제발ㅠ 많이 연습하자~~~