[백준/자바] 스택 - stack

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

10828번: 스택

 

풀이 방법

1. 입출력을 위해 Buffered를 사용한다.

2. 큐가 비었는지 확인은 boolean형을 사용하며 empty()값이 true면 비어있다는 뜻이다.

3. 각 조건에 맞게 if문을 작성한다. 

package stack;
 
import java.io.*;
import java.util.Stack;
 
 
public class 스택 {
     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());
         boolean sEmpty;
 
         Stack<Integer> stack = new Stack<>();
 
         for (int i = 0; i<N; i++){
             String input = br.readLine();
             sEmpty = stack.empty(); // true면 비어있다는 뜻이다.
 
             if(input.contains("push")){
                 String[] temp = input.split(" ");
                 stack.push(Integer.parseInt(temp[1]));
             } else if (input.contains("pop")){
                 if(sEmpty==true){
                     bw.write(-1+"\n");
                 } else {
                     bw.write(stack.pop()+"\n");
                 }
             } else if (input.contains("size")){
                 bw.write(stack.size()+"\n");
             } else if (input.contains("empty")){
                 if (sEmpty==true){
                     bw.write(1+"\n");
                 } else{
                     bw.write(0+"\n");
                 }
             } else if (input.contains("top")){
                 if (sEmpty==true){
                     bw.write(-1+"\n");
                 } else{
                     bw.write(stack.peek()+"\n");
                 }
 
             }
         }
         bw.close();
         br.close();
     }
}
 

 

풀이 후기

1. switch 를 사용하면 더 가독성 좋은 코드를 작성할 수 있을 거 같다.

2. Stack 라이브러리를 사용하였는데, 공부를 목적이라면 직접 메서드를 구현해보는 게 좋을 거 같다.