Categories: 백준

백준 10845 큐 (자바 Java)

java

백준 10845 큐

문제 정보

문제 링크

문제 출처 : 백준 Baekjoon
문제 번호 : 10845
문제 제목 : 큐
문제 난이도 : 실버4
풀이 언어 : Java

문제

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

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

접근

Scanner와 System.out.println은 시간 초과가 뜨기 때문에 BufferedReader와 BufferedWrite를 사용해 문제를 풀었다.

풀이

문자열 입력을 받기 위해 BufferedReader를 사용한다.
실행 결과를 출력하기 위해 BufferedWriter를 사용한다.
사용자로부터 입력 받은 첫 번째 줄(명령어의 수)을 정수로 변환한다.
정수를 저장하기 위한 자료구조로 Queue를 사용하고, 이를 LinkedList로 초기화한다.
입력 받은 명령어의 수만큼 반복하며, 각 줄의 명령어를 처리한다.
push 명령어 처리: 큐에 새로운 요소를 추가한다. 추가된 값은 lastValue 변수에 저장된다.
pop 명령어 처리: 큐에서 가장 앞에 있는 요소를 제거하고 그 값을 출력한다. 큐가 비었으면 -1을 출력한다.
size 명령어 처리: 큐에 저장된 요소의 수를 출력한다.
empty 명령어 처리: 큐가 비어있는지 확인하고 그 결과를 출력한다. 비어있으면 1, 그렇지 않으면 0을 출력한다.
front 명령어 처리: 큐의 가장 앞에 있는 요소를 출력한다. 큐가 비었으면 -1을 출력한다.
back 명령어 처리: 큐의 마지막에 추가된 요소(lastValue)를 출력한다. 큐가 비었으면 -1을 출력한다.

후기

큐를 활용한 기본적인 문제이다. 큐와 관련된 명령어들을 잘 숙지해야겠다.

코드

Java

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.BufferedWriter;
import java.io.OutputStreamWriter;
import java.io.IOException;
import java.util.LinkedList;
import java.util.Queue;

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));
        Queue<Integer> queue = new LinkedList<>();
        int lastValue = 0;

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

        for (int i = 0; i < N; i++) {
            String command = br.readLine();
            
            if (command.startsWith("push")) {
                int value = Integer.parseInt(command.substring(5));
                queue.add(value);
                lastValue = value;
            } else if (command.equals("pop")) {
                if (queue.isEmpty()) {
                    bw.write("-1\n");
                } else {
                    bw.write(queue.poll() + "\n");
                }
            } else if (command.equals("size")) {
                bw.write(queue.size() + "\n");
            } else if (command.equals("empty")) {
                bw.write((queue.isEmpty() ? 1 : 0) + "\n");
            } else if (command.equals("front")) {
                if (queue.isEmpty()) {
                    bw.write("-1\n");
                } else {
                    bw.write(queue.peek() + "\n");
                }
            } else if (command.equals("back")) {
                if (queue.isEmpty()) {
                    bw.write("-1\n");
                } else {
                    bw.write(lastValue + "\n");
                }
            }
        }
        br.close();
        bw.flush();
        bw.close();
    }
}