[Java] BOJ 10845 큐

문제 

문제 링크 https://www.acmicpc.net/problem/10845

• 정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
  • push X: 정수 X를 큐에 넣는 연산이다.
  • pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
  • size: 큐에 들어있는 정수의 개수를 출력한다.
  • empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
  • front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
  • back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
• 명령은 총 여섯 가지이다.
• 첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다.
• 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
• 출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.

아이디어

• ArrayDeque로 stack과 queue 기능을 한꺼번에 썼다.
• stack은 선입후출이고 queue는 선입선출이기 때문에 값을 꺼내는 방법이 다른데, front를 입력받을 때, 가장 앞에 있는 정수를 출력해야되고 back일때, 가장 뒤에 있는 정수를 출력해야되므로 ArrayDeque를 썼다. 
• 입력은 addLast로 queue 처럼 입력을 받고 값을 빼낼때는 pollFirst를 써서 가장 앞에 있는 정수를 뺐다.
• 사이즈는 size로 출력하고 가장 앞에 있는 값을 출력하기 위해 peekFirst를 썼다.
• 마지막으로 가장 뒤에 있는 값을 출력하기 위해 peekLast를 썼다. 

겪은 시행착오

• 처음에는 Queue로 풀려고 하다가 가장 뒤에 있는 값을 어떻게 출력하지란 문제점이 생겨 찾아보다가 ArrayDeque로 풀게 되었다... 새롭게 문제를 풀 수 있어서 좋았다..!

앞으로도 파이팅!

코드

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

public class BOJ10845 {
    public static void main(String[] args) throws Exception{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

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

        // queue와 stack을 모두 쓰기 위해서
        ArrayDeque<Integer> q = new ArrayDeque<>();

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

            // 명령어를 입력받음
            String order = st.nextToken();

            // 만약 명령어가 push라면
            if(order.equals("push")){
                int n = Integer.parseInt(st.nextToken());

                // 수가 queue처럼 저장되어야 하므로 addLast로
                // n을 queue에 추가
                q.addLast(n);
            }
            // 만약 명령어가 pop이라면
            else if(order.equals("pop")){
                // 만약 queue가 비어있다면 -1 출력
                if(q.isEmpty()){
                    sb.append(-1 + "\n");
                }
                // 비어있지 않다면
                else{
                    // 가장 위에 있는 정수를 빼내고 저장하고 출력
                    int n = q.pollFirst();

                    sb.append(n + "\n");
                }
            }
            // 만약 명령어가 size라면
            else if(order.equals("size")){
                // queue의 사이즈를 저장 후 출력
                int n = q.size();

                sb.append(n + "\n");
            }
            // 만약 명령어가 empty라면
            else if(order.equals("empty")){
                // queue가 비어있을 경우
                if(q.isEmpty()){
                    // 1출력
                    sb.append(1 + "\n");
                }
                // 비어있지 않다면
                else{
                    // 0출력
                    sb.append(0 + "\n");
                }
            }
            // 만약 명령어가 front이고
            else if(order.equals("front")){
                // queue가 비어있다면
                if(q.isEmpty()){
                    sb.append(-1 + "\n");
                }
                // 비어있지 않다면
                else{
                    // 제일 앞에 저장된 값을 저장 후 출력
                    int n = q.peekFirst();

                    sb.append(n + "\n");
                }
            }
            // 만약 명령어가 back이고
            else if(order.equals("back")){
                // queue가 비어있다면
                if(q.isEmpty()){
                    sb.append(-1 + "\n");
                }
                // 비어있지 않다면
                else{
                    // 제일 나중에 저장된 값을 저장 후 출력
                    int n = q.peekLast();

                    sb.append(n + "\n");
                }
            }
        }
        System.out.println(sb);
    }
}