[ baekjoon ] 10828. 스택

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 다섯 가지이다.
push X: 정수 X를 스택에 넣는 연산이다.
pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
size: 스택에 들어있는 정수의 개수를 출력한다.
empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

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

출력

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

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

public class Main { // 시간 초과 case
    public static void main(String[] args) {
        // 스캐너
        Scanner sc = new Scanner(System.in);

        // 명령의 수 num
        int num = sc.nextInt();
        sc.nextLine();

        Stack<Integer> stack = new Stack<>();
        String[] inputs = new String[num];
        for(int i = 0; i < num; i++) {
            inputs[i] = sc.nextLine();
        }

        for(int i = 0; i < num; i++){
            if(inputs[i].contains("push")){ // push
                stack.push(Integer.parseInt(inputs[i].split(" ")[1]));
            }
            else if(inputs[i].equals("top")){
                System.out.println(stack.isEmpty()?-1:stack.peek());
            }
            else if(inputs[i].equals("pop")){
                System.out.println(stack.isEmpty()?-1:stack.pop());
            }
            else if(inputs[i].equals("empty")){
                System.out.println(stack.isEmpty()?1:0);
            }
            else if(inputs[i].equals("size")){
                System.out.println(stack.size());
            }
            else{
                return;
            }
        }
    }
}

BUT 이 방식은 시간 초과로 인해 실패한다.
System.out.println()으로 매번 출력하면 시간초과

제한 시간이 0.5초(500ms)이하여야 한다.

그래서 Scanner와 StringBuilder를 사용하여 다시 코드를 작성하였다
나는 스택을 배열이 아닌, Stack 클래스를 사용하였다
cf)
stack을 int[] stack 배열로 선언해서, push, pop, size, empty, top(==peek)메서드를 각각 구현해도 된다

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

public class Main { // 스택 클래스 이용, StringBuilder

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

    int num = sc.nextInt();

    for(int i = 0; i < num; i++){
        String input = sc.next();

        switch (input) {
            case "push":
                stack.push(sc.nextInt());
                break;
            case "pop":
                sb.append(stack.isEmpty() ? -1 : stack.pop()).append("\n");
                break;
            case "top":
                sb.append(stack.isEmpty() ? -1 : stack.peek()).append("\n");
                break;
            case "empty":
                sb.append(stack.isEmpty() ? 1 : 0).append("\n");
                break;
            case "size":
                sb.append(stack.size()).append("\n");
                break;
        }
    }
    System.out.println(sb);
    }
}

스택 배열 vs 스택 클래스

메모리 - 배열 < 클래스
시간 - 배열 > 클래스