[백준_11000번] 강의실 배정

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

11000번: 강의실 배정

[문제]

 수강신청의 마스터 김종혜 선생님에게 새로운 과제가 주어졌다.  김종혜 선생님한테는 Si에 시작해서 Ti에 끝나는 N개의 수업이 주어지는데, 최소의 강의실을 사용해서 모든 수업을 가능하게 해야 한다.  참고로, 수업이 끝난 직후에 다음 수업을 시작할 수 있다. (즉, Ti ≤ Sj 일 경우 i 수업과 j 수업은 같이 들을 수 있다.) 수강신청 대충한 게 찔리면, 선생님을 도와드리자!

 

[입력 조건]

• 첫 번째 줄에 N이 주어진다. (1 ≤ N ≤ 200,000)
• 이후 N개의 줄에 Si, Ti가 주어진다. (0 ≤ Si < Ti ≤ 109)

 

[코드]

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

public class BaekJoon_11000 {
	
	public static void main(String[] args) throws IOException  {
		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());
			
			time[i][0]=Integer.parseInt(st.nextToken()); // 시작 시간
			time[i][1]=Integer.parseInt(st.nextToken()); // 종료 시간
		}
		
		Arrays.sort(time,new Comparator<int[]>() {
			@Override
			public int compare(int[] o1, int[] o2) {
				// 시작 시간이 같다면 종료 시간이 이른 순서대로 정렬
				if(o1[0]==o2[0])
					return Integer.compare(o1[1],o2[1]);
				
				// 기본적으로 시작 시간이 빠른 순서대로 정렬
				return Integer.compare(o1[0],o2[0]);
			}
		});
		
		PriorityQueue<Integer> pq=new PriorityQueue<>();
		pq.add(time[0][1]); // 첫 번째 수업의 종료 시간 저장
		
		// 두 번째 수업부터 탐색
		for(int i=1;i<n;i++) {
			 /*
			  * 다음 수업 시작시간이 이전 수업의 종료시간보다 크다면 큐에있는 수업 삭제 
			  * == 같은 강의실 사용 가능
			  */
			if(time[i][0]>=pq.peek())
				pq.poll();
			
			// 현재 수업의 종료 시간을 큐에 삽입
			pq.add(time[i][1]);
		}
		
		System.out.println(pq.size()); // 큐에 남아있는 데이터의 개수가 필요한 강의실의 수
		br.close();
	}

}

 

[고찰]

 이번 문제는 이전에 풀어봤던 백준 1931번 회의실 배정 문제랑 비슷한 유형이었다. 하지만 이번 문제는 우선순위 큐 카데고리에 있어 좀 더 쉽게 어떤 자료구조를 사용해야하는지 파악할 수 있었다. 그렇지 않았다면 시간초과 문제로 꽤 어렵게 느껴졌을 것 같다. 

 

*** 1) 수업시간 저장

 우선 n*2 크기의 2차원 배열에 수업의 시작 시간과 종료 시간을 저장하고 시작 시간이 빠른 순서대로 정렬한다. 이때 시작 시간이 같다면 종료 시간이 빠른 순서대로 정렬한다.

 

*** 2) 수업시간 비교

 2-1) 이후부터 우선순위 큐를 사용해야한다. 우선순위 큐에는 수업의 종료 시간을 저장한다. 첫 번째 수업의 종료 시간을 먼저 삽입한 뒤 두 번째 수업부터 탐색을 시작한다.

 2-2) 다음 수업의 시작 시간이 큐에 저장된 이전 수업의 종료 시간보다 늦다면 두 수업은 같은 강의실을 사용해도 되는 경우이기 때문에 큐에서 이전 수업의 종료 시간을 삭제한다. 

 2-3) 매 경우마다 수업의 종료 시간을 큐에 삽입한다. (강의 진행 의미)

 2-4) 모든 과정이 끝난 후 큐에 남아있는 데이터의 개수가 필요한 강의실의 수가 된다.

 

 이번 문제에서는 우선순위 큐를 사용해야 한다는 점과 모든 과정이 끝나고 큐에 남아있는 데이터의 개수가 필요한 강의식의 개수라는 것을 파악하는 점이 핵심이었던것 같다. 이번 코딩스터디 과정에서 평소에는 잘 풀어보지 않았던 우선순위 큐를 사용하여 해결해야 하는 문제들을 풀어보고 있는데 많은 공부가 되는것 같다.