[백준] 18258번 '큐2' (자바/JAVA)

문제

정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 여섯 가지이다.

• push X: 정수 X를 큐에 넣는 연산이다.
• pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• size: 큐에 들어있는 정수의 개수를 출력한다.
• empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
• front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 2,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

예제 입력 1 

15
push 1
push 2
front
back
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
front

예제 출력 1 

1
2
2
0
1
2
-1
0
1
-1
0
3

코드 StringBuilder 사용안 한 버전

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

class Main {
    public static void main(String[] args) throws IOException {
        // BufferedReader를 사용하여 입력 스트림을 효율적으로 읽어옵니다.
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        // 첫 번째 줄에서 명령의 개수 N을 읽어옵니다.
        int N = Integer.parseInt(br.readLine());
        // 정수를 저장할 큐(Queue)를 LinkedList로 구현합니다. LinkedList는 Queue 인터페이스를 구현하며, FIFO (First-In, First-Out) 동작을 지원합니다.
        Queue<Integer> queue = new LinkedList<>();

        // N번만큼 명령을 처리합니다.
        for (int i = 0; i < N; i++) {
            // 각 명령이 포함된 한 줄을 읽어옵니다.
            String line = br.readLine();
            // 읽어온 줄을 공백을 기준으로 분리하여 명령어와 값을 배열 'parts'에 저장합니다.
            String[] parts = line.split(" ");
            // 배열의 첫 번째 요소는 명령어입니다.
            String command = parts[0];

            // 명령어에 따라 큐 연산을 수행합니다.
            switch (command) {
                // "push" 명령어인 경우
                case "push":
                    // 공백으로 분리된 배열의 두 번째 요소는 정수 값입니다. 이를 Integer로 변환하여 큐에 추가합니다.
                    int value = Integer.parseInt(parts[1]);
                    queue.offer(value); // offer()는 큐에 요소를 추가하고 성공하면 true를 반환합니다.
                    break;
                // "pop" 명령어인 경우
                case "pop":
                    // 큐가 비어있는지 확인합니다.
                    if (queue.isEmpty()) {
                        // 비어있으면 -1을 출력합니다.
                        System.out.println(-1);
                    } else {
                        // 비어있지 않으면 큐의 가장 앞에 있는 요소를 제거하고 출력합니다.
                        System.out.println(queue.poll()); // poll()은 큐에서 요소를 제거하고 반환하며, 비어있으면 null을 반환합니다.
                    }
                    break;
                // "size" 명령어인 경우
                case "size":
                    // 큐에 들어있는 요소의 개수를 출력합니다.
                    System.out.println(queue.size());
                    break;
                // "empty" 명령어인 경우
                case "empty":
                    // 큐가 비어있으면 1, 아니면 0을 출력합니다.
                    System.out.println(queue.isEmpty() ? 1 : 0);
                    break;
                // "front" 명령어인 경우
                case "front":
                    // 큐가 비어있는지 확인합니다.
                    if (queue.isEmpty()) {
                        // 비어있으면 -1을 출력합니다.
                        System.out.println(-1);
                    } else {
                        // 비어있지 않으면 큐의 가장 앞에 있는 요소를 반환하고 출력합니다. (제거하지 않습니다.)
                        System.out.println(queue.peek()); // peek()은 큐의 맨 앞 요소를 반환하며, 비어있으면 null을 반환합니다.
                    }
                    break;
                // "back" 명령어인 경우
                case "back":
                    // 큐가 비어있는지 확인합니다.
                    if (queue.isEmpty()) {
                        // 비어있으면 -1을 출력합니다.
                        System.out.println(-1);
                    } else {
                        // LinkedList로 캐스팅하여 큐의 가장 뒤에 있는 요소를 얻어옵니다. Queue 인터페이스는 직접적으로 뒤쪽 요소 접근을 제공하지 않습니다.
                        LinkedList<Integer> list = (LinkedList<Integer>) queue;
                        System.out.println(list.getLast());
                    }
                    break;
            }
        }
        // BufferedReader를 닫아 사용한 시스템 자원을 해제합니다.
        br.close();
    }
}

코드 StringBuilder사용 버전

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

class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        Queue<Integer> queue = new LinkedList<>();
        StringBuilder sb = new StringBuilder(); 

        for (int i = 0; i < N; i++) {
            String line = br.readLine();
            String[] parts = line.split(" ");
            String command = parts[0];

            switch (command) {
                case "push":
                    int value = Integer.parseInt(parts[1]);
                    queue.offer(value);
                    break;
                case "pop":
                    if (queue.isEmpty()) {
                        sb.append("-1\n");
                    } else {
                        sb.append(queue.poll()).append("\n");
                    }
                    break;
                case "size":
                    sb.append(queue.size()).append("\n");
                    break;
                case "empty":
                    sb.append(queue.isEmpty() ? 1 : 0).append("\n");
                    break;
                case "front":
                    if (queue.isEmpty()) {
                        sb.append("-1\n");
                    } else {
                        sb.append(queue.peek()).append("\n");
                    }
                    break;
                case "back":
                    if (queue.isEmpty()) {
                        sb.append("-1\n");
                    } else {
                        LinkedList<Integer> list = (LinkedList<Integer>) queue;
                        sb.append(list.getLast()).append("\n");
                    }
                    break;
            }
        }
        System.out.println(sb.toString());
        br.close();
    }
}