본문 바로가기
공부 정리/백준

[백준] 자바 18258 큐 2

by 경적필패. 2021. 11. 26.
반응형

문제

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

명령은 총 여섯 가지이다.

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

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 2,000,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.


테스트 케이스

 

입력 1

15
push 1
push 2
front
back
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
front

출력 1

1
2
2
0
1
2
-1
0
1
-1
0
3

 

 


접근

Deque와 스위치를 이용하니 아주 간단하게 해결되었습니다.


코드

import java.util.*;
import java.util.stream.IntStream;
import java.awt.Point;
import java.io.*;
	
	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 N = Integer.parseInt(br.readLine());
	    	Deque<Integer> q = new LinkedList<>();
	    	
	    	for(int i=0; i<N; i++) {
	    		StringTokenizer st = new StringTokenizer(br.readLine());
	    		String temp = st.nextToken();
	    		switch(temp) {
	    		
	    		case "push": q.add(Integer.parseInt(st.nextToken())); break;
	    		case "pop": 
	    			if(q.isEmpty()) {
	    			bw.write(String.valueOf(-1)+"\n");
	    			}
	    			else {
	    			bw.write(String.valueOf(q.poll())+"\n");
	    			}
	    			break;
	    		case "size": bw.write(String.valueOf(q.size())+"\n"); break;
	    		case "empty": 
	    			if(q.isEmpty()) bw.write("1"+"\n");
	    			else bw.write("0"+"\n");
	    			break;
	    		case "front":
	    			if(q.isEmpty()) bw.write("-1"+"\n");
	    			else bw.write(String.valueOf(q.peekFirst())+"\n");
	    			break;
	    		case "back":
	    			if(q.isEmpty()) bw.write("-1"+"\n");
	    			else bw.write(String.valueOf(q.peekLast())+"\n");
	    			break;
	    		}
	    	}
	    	br.close();
	    	bw.flush();
	    	bw.close();
	    }
	}

주의

X

 

반응형

댓글