[99클럽] 20일차 문제: 회전초밥

 

 

백준 회전초밥 문제. 이제 문제의 답이 길어지고 복잡해지면서 gpt에 의존하고 있다....

해시맵의 putIfAbsent() 메소드는 잘 사용해 보지 않았는데 다음기회에 사용해보면 좋을 것 같다. 

 

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));
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken()); // 손님 수
        int M = Integer.parseInt(st.nextToken()); // 초밥 수

        // 초밥을 원하는 손님 리스트
        Map<Integer, Queue<Integer>> sushiToCustomers = new HashMap<>();
        int[] sushiCount = new int[N]; // 손님별로 먹은 초밥 개수
        Set<Integer>[] customerOrders = new HashSet[N]; // 손님별 주문 목록

        // 손님들의 주문 목록을 저장
        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());
            int k = Integer.parseInt(st.nextToken()); // 주문한 초밥 개수
            customerOrders[i] = new HashSet<>();

            for (int j = 0; j < k; j++) {
                int sushiType = Integer.parseInt(st.nextToken());
                customerOrders[i].add(sushiType); // 손님의 주문 목록에 추가

                // 초밥별로 해당 초밥을 원하는 손님을 큐에 저장
                sushiToCustomers.putIfAbsent(sushiType, new LinkedList<>());
                sushiToCustomers.get(sushiType).add(i);
            }
        }

        // 초밥을 만들면서 손님에게 전달
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < M; i++) {
            int sushi = Integer.parseInt(st.nextToken());

            // 초밥을 원하는 손님이 존재하면
            if (sushiToCustomers.containsKey(sushi)) {
                Queue<Integer> queue = sushiToCustomers.get(sushi);

                while (!queue.isEmpty()) {
                    int customer = queue.poll(); // 초밥을 원하는 첫 번째 손님 선택

                    // 만약 손님이 이미 해당 초밥을 먹었다면 패스
                    if (!customerOrders[customer].contains(sushi)) continue;

                    // 손님이 초밥을 먹음
                    sushiCount[customer]++;
                    customerOrders[customer].remove(sushi); // 한 번 먹었으니 삭제
                    break; // 초밥을 먹었으면 종료 (다음 손님은 받을 수 없음)
                }
            }
        }

        // 결과 출력
        StringBuilder sb = new StringBuilder();
        for (int count : sushiCount) {
            sb.append(count).append("\n");
        }
        System.out.print(sb);
    }
}