[백준] 큐, 덱 - 18258번 큐2 java 자바

문제

https://www.acmicpc.net/problem/18258

18258번: 큐 2

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

명령은 총 여섯 가지이다.

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

풀이

그냥 큐 기본 문제 같은데 왜 정답률이 생각보다 낮지? 했는데 역시나 Scanner 와 System.out.println으로 풀면 시간초과가 나는 문제였다. 나는 LinkedList 라이브러리를 사용했기 때문에 peek, isEmpty, poll, offer 등을 이용하였고 가장 뒤에 있는 정수를 출력하는 내장함수는 없기 때문에 offer할 때마다 last라는 별도의 변수에 값을 담아주었다.(큐는 FIFO, 선입선출, 즉 가장 최근에 들어온 게 가장 뒤에 줄을 선다.)

 

코드

package 큐_덱;

import java.util.*;
import java.io.*;
public class No18258_큐2 {

	public static void main(String[] args)throws IOException{
		// TODO Auto-generated method stub
		BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb=new StringBuilder();
		
		
		int N=Integer.parseInt(br.readLine());
		StringTokenizer command;
		Queue<Integer> q=new LinkedList<>();
		
		int last=0;

		for(int i=0;i<N;i++) {
			command=new StringTokenizer(br.readLine()," ");
			switch(command.nextToken()) {
			case "push":
				int num=Integer.parseInt(command.nextToken());
				last=num;
				q.offer(num);
				break;
			case "front":
				if(!q.isEmpty())
					sb.append(q.peek()).append('\n');
				else
					sb.append(-1).append('\n');
				break;
			case "back":
				if(!q.isEmpty()) 
					sb.append(last).append('\n');
				else
					sb.append(-1).append('\n');
				break;
			case "size":
				sb.append(q.size()).append('\n');
				break;
			case "empty":
				if(q.isEmpty())
					sb.append(1).append('\n');
				else
					sb.append(0).append('\n');
				break;
			case "pop":
				if(!q.isEmpty())
					sb.append(q.poll()).append('\n');
				else
					sb.append(-1).append('\n');
				break;
			}
		}
		
		System.out.println(sb);
	}

}