본문 바로가기
코딩테스트 준비/백준

[백준] 18258번 - 큐 2 [Java]

by mwzz6 2025. 1. 12.

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

 

[백준] 18258번 - 큐 2 [Java]
[백준] 18258번 - 큐 2 [Java]


1.  아이디어

 

기존 큐 문제에서 명령의 수만 늘어난 문제로 동일한 코드로 해결했다. 명령의 수가 제법 많아서 BufferedWriter를 사용하는 것을 추천한다.([코딩테스트 준비/백준] - [백준] 10845번 - 큐 [Java])


2. 문제풀이

 

동일하게 구현했다.


3. 코드

 

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();
        StringTokenizer st;

        Deque<Integer> q = new ArrayDeque<>();

        int N = Integer.parseInt(br.readLine());

        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());

            String order = st.nextToken();

            switch (order) {
                case "push":
                    int X = Integer.parseInt(st.nextToken());
                    q.offer(X);
                    break;

                case "pop":
                    if (q.isEmpty()) sb.append(-1).append("\n");
                    else sb.append(q.poll()).append("\n");
                    break;

                case "size":
                    sb.append(q.size()).append("\n");
                    break;

                case "empty":
                    int flag = q.isEmpty() ? 1 : 0;
                    sb.append(flag).append("\n");
                    break;

                case "front":
                    if (q.isEmpty()) sb.append(-1).append("\n");
                    else sb.append(q.peekFirst()).append("\n");
                    break;

                case "back":
                    if (q.isEmpty()) sb.append(-1).append("\n");
                    else sb.append(q.peekLast()).append("\n");
                    break;
            }
        }

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

4. 후기