본문 바로가기

생각정리/코딩테스트

[JAVA 알고리즘]BAEKJOON 18258번 큐 2

큐 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가 있는 걸 확인을 했었는데 버퍼라이터의 경우는 일일이 줄 바꿈을 해주어야 돼서 그냥 계속 프린터라인을 사용했는데 코드가 같더라도 한 번에 출력하는 부분에 있어서 시간이 많이 차이나는 걸 알았다. 프린터라인일 때는 시간초과로 실패하던 게 버퍼라이터로 바꾸니 바로 성공.. 입력문 출력문도 하나만 사용하는 게 아니라 각 상황에 맞게 잘 사용해야겠다.