[99클럽] 14일차 문제: 식당 메뉴

 

 

 

오늘을 백준 식당메뉴 문제였는데 문제가 이해하기 너무 어려웠다. 그래서 답안을 참고했는데 읽고 나니 어려운 문제는 아니였는데 문자열 입력 처리를 해야 해서 더 복잡해진 것 같다. 코드도 길어지고. 밑에 코드는 GPT를 도움을 받아 작성했다.

 

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();

        int n = Integer.parseInt(br.readLine()); // 정보의 개수

        List<Integer> listA = new ArrayList<>(); // 원하는 걸 먹은 학생
        List<Integer> listB = new ArrayList<>(); // 원하지 않는 걸 먹은 학생
        List<Integer> listC = new ArrayList<>(); // 못 먹은 학생

        Deque<int[]> queue = new LinkedList<>(); // 식당 대기 줄 (FIFO)

        for (int i = 0; i < n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int type = Integer.parseInt(st.nextToken());

            if (type == 1) { // 학생 도착
                int student = Integer.parseInt(st.nextToken());
                int favoriteMenu = Integer.parseInt(st.nextToken());
                queue.offer(new int[]{student, favoriteMenu});
            } else { // 음식 준비됨
                int menuReady = Integer.parseInt(st.nextToken());

                if (!queue.isEmpty()) {
                    int[] student = queue.poll(); // 대기열의 첫 번째 학생

                    if (student[1] == menuReady) { // 원하는 메뉴를 먹음
                        listA.add(student[0]);
                    } else { // 원하지 않는 메뉴를 먹음
                        listB.add(student[0]);
                    }
                }
            }
        }

        // 줄에 남아 있는 학생들은 먹지 못한 학생들
        while (!queue.isEmpty()) {
            listC.add(queue.poll()[0]);
        }

        // 정렬 후 출력
        printList(sb, listA);
        printList(sb, listB);
        printList(sb, listC);

        bw.write(sb.toString());
        bw.flush();
        bw.close();
        br.close();
    }

    // 리스트 정렬 및 출력 함수
    private static void printList(StringBuilder sb, List<Integer> list) {
        if (list.isEmpty()) {
            sb.append("None\n");
        } else {
            Collections.sort(list);
            for (int num : list) {
                sb.append(num).append(" ");
            }
            sb.append("\n");
        }
    }
}