[Algorithm] 백준 17178 줄서기

[ 문제 ]

아이즈원의 팬인 시온이는 드디어 티켓팅에 성공하여 콘서트를 갔다. 콘서트장에 일찍 도착한 시온이는 기대하며 입장을 위해 줄을 섰다.

하지만 아이즈원의 인기대로 시온이를 포함한 많은 팬이 줄을 서고 있다.

콘서트의 입장이 시작되었고 입장은 티켓 번호 순서대로 이루어졌다.

하지만 입구에 너무 많은 팬이 몰려 아무도 이동할 수 없는 상황이 되었고,

결국 주최 측에서 인원을 정렬시켜 다음과 같이 간신히 사람 한 줄이 설 수 있는 대기 공간을 만들었다.

 

주최 측은 번호표 순서대로만 통과할 수 있는 입구를 만들어 두었지만, 줄에서는 마구잡이로 사람들이 기다리고 있다.

대기 공간을 이용하여 입장이 원활히 이루어지도록 하려고 한다.

콘서트장에 사람들이 제대로 들어갈 수 있는지 확인해보자.

 

사람들은 현재 5명씩 N 줄을 서 있고, 첫 번째 줄 맨 앞사람만 이동이 가능하다.

이 사람은 콘서트장으로 입장할 수도 있고 대기 공간에서 다시 기다릴 수도 있다.

한 줄의 사람이 다 이동했다면 그다음 줄의 사람들이 이동한다.

대기 공간에는 한 줄로만 설 수 있는 공간이 있으며, 마지막에 들어온 사람부터 나갈 수 있다.

나갈 경우 바로 입장해야 하고, 다시 줄로 돌아갈 수는 없다.

티켓은 A-123과 같이 한 개의 대문자 알파벳과 '-', 1000 미만의 자연수의 조합으로 이뤄어져 있다.

만약 수가 7이라면 A-7과 같이 주어진다.

티켓의 순서는 알파벳이 빠른 티켓이 빠르며, 동일하다면 더 적은 수가 더 빠르다. 티켓 번호는 중복되게 주어지지 않는다.

위의 예시를 예로 들면 다음과 같이 모든 사람들이 입장할 수 있다.

그림과는 달리 대기 공간에는 무한히 많은 사람들이 들어올 수 있다는 것에 주의하여야 한다.

 

[ 입력 ]

• 첫째 줄에는 줄에서 기다리고 있는 사람들의 줄 수 N이 주어진다. (1 ≤ N ≤ 100)
• 둘째 줄부터 N 개의 줄에는 한 줄에 서 있는 5명의 티켓 번호가 주어진다.
• 사람들이 서 있는 순서대로 입력이 주어진다.

 

[ 출력 ]

• 모든 사람이 무사히 콘서트장에 입장할 수 있다면 “GOOD”을 출력하고 그렇지 않다면 “BAD”를 출력한다.

 

---

티켓을 우선순위대로 정렬하기 위해 PriorityQueue를 사용하였습니다.

알파벳이 같은 경우 숫자가 작은 순서대로 정렬을 해야 하는데

기본 Comparable을 사용하면 숫자를 문자열로 인식해서 결과가 다르게 나올 수 있습니다.

예를 들어, "G-104"와 "G-4" 중 "G-4"의 우선순위가  높아야 하지만 기본 Comparable을 사용하면 "G-104"의 우선순위가 높습니다.

따라서 Comparable의 compareTo 메소드를 재정의 했습니다.

 

기본적으로 줄 서 있는 사람들의 티켓 정보를 Queue에 저장했고,

대기 장소에 줄 서 있는 사람은 마지막 사람이 나갈 수 있으므로 Stack을 사용했습니다.

 

모든 티켓의 우선순위를 확인할 때까지 반복문을 실행할 것입니다.

현재 들어가야 하는 사람이 줄 맨 앞 사람도, 대기 공간 마지막 사람도 아닌 경우 대기 공간에 그 사람을 보냈습니다.

 

그렇게 모든 티켓의 우선순위를 확인했는데

대기 공간에 사람이 남아있다면 입장을 못하고 있는 경우라고 판단하여 "BAD"를,

대기 공간에도 사람이 없다면 "GOOD"를 출력해줬습니다. 

 

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

public class Main {
	
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st;
		
		int N = Integer.parseInt(br.readLine()); // 줄에서 기다리고 있는 사람들의 줄 수
		PriorityQueue<String> order = new PriorityQueue<>(new Comparator<String>() { // 모든 사람의 티켓
			public int compare(String o1, String o2) {
				String[] ticket1 = o1.split("-");
				String[] ticket2 = o2.split("-");
				
				if(!ticket1[0].equals(ticket2[0])) return ticket1[0].charAt(0) - ticket2[0].charAt(0); // 알파벳 순서대로
				else return Integer.parseInt(ticket1[1]) - Integer.parseInt(ticket2[1]); // 알파벳이 같다면 숫자가 작은 순서대로
			}
		});  
		Queue<String> line = new LinkedList<>(); // 줄
		Stack<String> wait = new Stack<>(); // 대기 공간
		
		for(int n = 0; n < N; n++) {
			st = new StringTokenizer(br.readLine());
			
			for(int i = 0; i < 5; i++) {
				String ticket = st.nextToken();
						
				line.offer(ticket);
				order.offer(ticket);
			}
		}
		
		while(!order.isEmpty()) {
			String cur = order.poll(); // 들어가야 하는 사람의 티켓
			
			while((!line.isEmpty() && !cur.equals(line.peek())) && ((!wait.isEmpty() && !cur.equals(wait.peek())) || wait.isEmpty())) { // 들어가야 하는 사람이 줄 맨 앞 사람도 아니고 대기 공간 마지막 사람도 아닌 경우 
				wait.add(line.poll()); // 대기 공간으로 이동
			}
			
			if(cur.equals(line.peek())) // 들어갈 차례인 사람이 줄 맨 앞에 있는 경우
				line.poll();
			else if(!wait.isEmpty() && cur.equals(wait.peek())) // 대기 공간에 사람이 있고, 들어갈 차례인 사람이 대기 공간 마지막에 있는 경우
				wait.pop();
		}
		
		if(wait.isEmpty()) System.out.println("GOOD");
		else System.out.println("BAD");
	}
}