백준 10828 스택(Java)

 

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

10828번: 스택

 

---

 

해석 및 팁

 

---

 

이 문제는 Stack의 기본을 알면 풀 수 있는 문제입니다. 예를 들어 pop이면 stack.pop(), push이면 stack.push(n)처럼 활용해 주면 됩니다 top 인경우에는 스택의 가장 위에 있는 정수를 출력하면 되므로 stack.peek()를 사용해주면 됩니다.

 

---

 

Java 코드

 

---

import java.util.Scanner;
import java.util.Stack;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    Stack<Integer> stack = new Stack<>();
    StringBuilder sb = new StringBuilder();

    int n = sc.nextInt();
    
    for(int i = 0; i < n; i++) {
      String str = sc.next();

      if(str.equals("push")) stack.push(sc.nextInt());
      else if(str.equals("pop")) {
        if(stack.empty()) sb.append(-1+"\n");
        else sb.append(stack.pop()+"\n");
      }
      else if(str.equals("size")) sb.append(stack.size()+"\n");
      else if(str.equals("empty")) {
        if(stack.empty()) sb.append(1+"\n");
        else sb.append(0+"\n");
      }
      else if(str.equals("top")) {
        if(stack.empty()) sb.append(-1+"\n");
        else sb.append(stack.peek()+"\n");
      }
      
    }

    System.out.println(sb);
  }
}