[JAVA] 백준 28278 스택 2

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

28278번: 스택 2

 

풀이 과정 

 

- 자료구조 전공책 보면서 직접 구현하려다가 컬렉션 프레임워크 Stack 사용 

 

Stack 사용법 

import java.util.Stack;

Stack<Integer> stackInt = new Stack<>(); //int형 스택 선언
Stack<String> stackString = new Stack<>(); //String형 스택 선언

stackInt.push(1); //stack에 값 1 추가 
stackInt.push(2); //stack에 값 2 추가 

stackInt.pop(); //stack에 값 제거 

stackInt.clear(); //stack의 전체 값 제거 (초기화)

stackInt.peek(); //stack의 가장 상단의 값 출력 

stackInt.size(); //stack의 크기 출력 

stackInt.empty(); //stack이 비어 있다면 true

stackInt.containse(1); // stack에 1이 있다면 true

 

 

정답

 

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;
import java.util.Stack;

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));
	   
	   
	   Stack<Integer>stack = new Stack<>();
	   
	   int n = Integer.parseInt(br.readLine());
	   
	   for(int i=0; i<n; i++)
	   {
		   String s = br.readLine();
		   StringTokenizer st = new StringTokenizer(s);
		   
		   switch(st.nextToken())
		   {
		   case "1": //푸시
			   stack.push(Integer.parseInt(st.nextToken()));
			   break;
		   case "2": //팝
			   if(stack.isEmpty())
			   {
				   bw.write(-1+"");
				   bw.newLine();
			   }   	
			   else
			   {
				   bw.write(stack.pop()+"");
				   bw.newLine();
			   }
			   break;
		   case "3": //스택 사이즈
			   bw.write(stack.size()+"");
			   bw.newLine();
			   break;
		   case "4": //스택이 비어 있는가?
			   if(stack.isEmpty())
			   {
				   bw.write(1+"");
				   bw.newLine();
			   }
			   else
			   {
				   bw.write(0+"");
				   bw.newLine();
			   }
			   break;
		   case "5": //피크
			   if(stack.isEmpty())
			   {
				   bw.write(-1+"");
				   bw.newLine();
			   }
			   else
			   {
				   bw.write(stack.peek()+"");
				   bw.newLine();
			   }
			   break;

		   }
	   }
	   
	   bw.flush();
	   bw.close();
	   
	   
   }
}