생각정리/코딩테스트
[JAVA 알고리즘]BAEKJOON 18258번 큐 2
생각중임
2023. 8. 1. 21:42
큐 2
문제 설명
정수를 저장하는 큐를 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 여섯 가지이다.
- push X: 정수 X를 큐에 넣는 연산이다.
- pop: 큐에서 가장 앞에 있는 정수를 빼고, 그 수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- size: 큐에 들어있는 정수의 개수를 출력한다.
- empty: 큐가 비어있으면 1, 아니면 0을 출력한다.
- front: 큐의 가장 앞에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
- back: 큐의 가장 뒤에 있는 정수를 출력한다. 만약 큐에 들어있는 정수가 없는 경우에는 -1을 출력한다.
제한사항
- 없음
입력
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 2,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
입출력 예
answers | return |
15 push 1 push 2 front back size empty pop pop pop size empty pop push 3 empty front |
1 2 2 0 1 2 -1 0 1 -1 0 3 |
나의 문제풀이
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayDeque;
import java.util.Deque;
import java.util.StringTokenizer;
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));
int T = Integer.parseInt(br.readLine());
Deque<String> queue = new ArrayDeque<>();
for (int i = 1; i <= T; i++) {
StringTokenizer command = new StringTokenizer(br.readLine());
switch (command.nextToken()) {
case "push":
queue.add(command.nextToken());
break;
case "pop":
if (queue.isEmpty()) {
bw.write("-1\n");
} else {
bw.write(queue.poll()+"\n");
}
break;
case "size":
bw.write(queue.size()+"\n");
break;
case "empty":
if (queue.isEmpty()) {
bw.write("1\n");
} else {
bw.write("0\n");
}
break;
case "front":
if (queue.isEmpty()) {
bw.write("-1\n");
} else {
bw.write(queue.peekFirst()+"\n");
}
break;
case "back":
if (queue.isEmpty()) {
bw.write("-1\n");
} else {
bw.write(queue.peekLast()+"\n");
}
break;
}
}
bw.flush();
bw.close();
}
}
- 양방향 입력 및 삭제를 하기 위해 덱을 이용한다.
- 명령어는 토큰을 이용해서 값을 받아서 사용한다.
- 케이스문을 통해서 명령어를 받아서 각 기능을 구현한다.
- 출력은 버퍼라이터를 이용해서 한 번에 출력한다.
느낀 점
버퍼리더를 찾아보면서 출력도 System.out.println와 BefferedWriter가 있는 걸 확인을 했었는데 버퍼라이터의 경우는 일일이 줄 바꿈을 해주어야 돼서 그냥 계속 프린터라인을 사용했는데 코드가 같더라도 한 번에 출력하는 부분에 있어서 시간이 많이 차이나는 걸 알았다. 프린터라인일 때는 시간초과로 실패하던 게 버퍼라이터로 바꾸니 바로 성공.. 입력문 출력문도 하나만 사용하는 게 아니라 각 상황에 맞게 잘 사용해야겠다.