백준 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();
}
}