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

🔗 백준 18258 - 큐 2

문제

알고리즘 분류

• 자료 구조
• 큐

---

풀이

1. 큐 선언 및 입력

• 큐 문제이기 때문에 처음에는 Queue<Integer> q = new LinkedList<>();와 같이 선언하려고 했다. 그러나 문제 조건을 보면 back 명령이 들어왔을 때 맨 뒤의 정수를 출력해야하는데, Queue 인터페이스는 이러한 메소드를 지원하지 않는다. 따라서 아래 코드와 같이 LinkedList로 선언하여 getFirst(), getLast() 메소드를 이용했다.
• LinkedList 말고 ArrayDeque를 사용해도 된다.

		LinkedList<Integer> q = new LinkedList<>();

        int n = Integer.parseInt(br.readLine());
        for(int i = 0; i<n; i++){
            String command = br.readLine();

2. 명령에 따른 조건문

            if (command.contains("push")){
                st = new StringTokenizer(command);
                command = st.nextToken();
                q.add(Integer.parseInt(st.nextToken()));
            } else if (command.equals("pop")){
                if (q.isEmpty())
                    sb.append(-1).append("\n");
                else sb.append(q.poll()).append("\n");
            } else if (command.equals("size")){
                sb.append(q.size()).append("\n");
            } else if (command.equals("empty")){
                if (q.isEmpty())
                    sb.append(1).append("\n");
                else sb.append(0).append("\n");
            } else if (command.equals("front")){
                if (q.isEmpty())
                    sb.append(-1).append("\n");
                else sb.append(q.getFirst()).append("\n");
            } else {
                if (q.isEmpty())
                    sb.append(-1).append("\n");
                else sb.append(q.getLast()).append("\n");
            }

전체 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        StringBuilder sb = new StringBuilder();
        LinkedList<Integer> q = new LinkedList<>();

        int n = Integer.parseInt(br.readLine());
        for(int i = 0; i<n; i++){
            String command = br.readLine();
            if (command.contains("push")){
                st = new StringTokenizer(command);
                command = st.nextToken();
                q.add(Integer.parseInt(st.nextToken()));
            } else if (command.equals("pop")){
                if (q.isEmpty())
                    sb.append(-1).append("\n");
                else sb.append(q.poll()).append("\n");
            } else if (command.equals("size")){
                sb.append(q.size()).append("\n");
            } else if (command.equals("empty")){
                if (q.isEmpty())
                    sb.append(1).append("\n");
                else sb.append(0).append("\n");
            } else if (command.equals("front")){
                if (q.isEmpty())
                    sb.append(-1).append("\n");
                else sb.append(q.getFirst()).append("\n");
            } else {
                if (q.isEmpty())
                    sb.append(-1).append("\n");
                else sb.append(q.getLast()).append("\n");
            }
        }
        System.out.println(sb);
    }
}

---