[programmers] 호텔 대실

문제
https://school.programmers.co.kr/learn/courses/30/lessons/155651

프로그래머스

 

---

< 호텔 대실 >

 

문제 풀이 (Java)

import java.util.*;

class Solution {
	public int solution(String[][] book_time) {
		int answer = 0;

		Arrays.sort(book_time, new Comparator<String[]>() {
			@Override
			public int compare(String[] o1, String[] o2) {

				int h1 = Integer.parseInt(o1[0].split(":")[0]);
				int m1 = Integer.parseInt(o1[0].split(":")[1]);

				int h2 = Integer.parseInt(o2[0].split(":")[0]);
				int m2 = Integer.parseInt(o2[0].split(":")[1]);

				if (h1 == h2) {
					return m1 - m2;
				}
				return h1 - h2;
			}
		});

		PriorityQueue<int[]> queue = new PriorityQueue<>(new Comparator<int[]>() {

			@Override
			public int compare(int[] o1, int[] o2) {
				if (o1[0] == o2[0]) {
					return o1[1] - o2[1];
				}
				return o1[0] - o2[0];
			}

		});

		for (int i = 0; i < book_time.length; i++) {
			int h1 = Integer.parseInt(book_time[i][0].split(":")[0]);
			int m1 = Integer.parseInt(book_time[i][0].split(":")[1]);

			int h2 = Integer.parseInt(book_time[i][1].split(":")[0]);
			int m2 = Integer.parseInt(book_time[i][1].split(":")[1]);

			if (!queue.isEmpty()) {
				if (queue.peek()[0] == h1) {
					if (queue.peek()[1] <= m1) {
						queue.poll();
					}
				} else if (queue.peek()[0] < h1) {
					queue.poll();
				}
			}
			m2 += 10;
			if (m2 >= 60) {
				m2 -= 60;
				h2 += 1;
			}
			queue.add(new int[] { h2, m2 });
			answer = Math.max(answer, queue.size());
		}

		return answer;
	}
}

 

먼저 예약 시각을 대실 시작 시각이 빠른 순대로 정렬한다. 그 후 PriorityQueue를 종료 시각이 빠른 순대로 정렬하도록 선언한다. 예약 시각을 탐색하며 PriorityQueue에 값이 있다면 첫 번째 값의 종료 시각과 예약 시각의 시작 시각을 비교해 시작 시각이 종료 시각보다 뒤라면 poll한 후 추가한다. 시작 시각이 종료 시각보다 앞이거나 PriorityQueue가 비어있다면 그냥 추가한다. 이때, PriorityQueue의 크기의 최댓값을 answer로 업데이트한다. 

 

최종 answer을 반환한다. 

 

* 다 풀고 보니 시각을 계속해서 h, m으로 나누는 것보다 m으로 통일해서 풀면 좋았겠다는 생각이 들었다. 

---

---

 

출처: 프로그래머스 코딩 테스트 연습, 
https://school.programmers.co.kr/learn/challenges