백준 JAVA 10828 스택

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

10828번: 스택

println으로 출력을 하면 시간초과로 실패된다.

StringBuilder를 사용해서 출력해야 한다.

 

package BAEKJOON.Silver.Ⅳ;

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

public class NO10828_2 {
    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") ) {
                int num = sc.nextInt();
                stack.push(num);
            } else if (str.equals("pop")) {
                if (stack.isEmpty()) {
                    sb.append(-1).append("\n");
                } else {
                    sb.append(stack.pop()).append("\n");
                }
            } else if (str.equals("size")) {
                sb.append(stack.size()).append("\n");
            } else if (str.equals("empty")) {
                if (!stack.isEmpty()) {
                    sb.append(0).append("\n");
                } else {
                    sb.append(1).append("\n");
                }
            } else if (str.equals("top")) {
                if (stack.isEmpty()) {
                    sb.append(-1).append("\n");
                } else {
                    sb.append(stack.peek()).append("\n");
                }
            }
        }
        System.out.println(sb);
    }
}

 

 

아래는 시간초과 실패코드

package BAEKJOON.Silver.Ⅳ;

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

public class NO10828 {
    public static void main(String[] args) {

        Scanner sc = new Scanner(System.in);
        Stack<Integer> stack = new Stack<>();
        int n = sc.nextInt();

        for (int i = 0; i < n; i++) {
            String str = sc.next();
            if (str.equals("push") ) {
                int num = sc.nextInt();
                stack.push(num);
            } else if (str.equals("pop")) {
                if (stack.isEmpty()) {
                    System.out.println(-1);
                } else {
                    System.out.println(stack.pop());
                }
            } else if (str.equals("size")) {
                System.out.println(stack.size());
            } else if (str.equals("empty")) {
                if (!stack.isEmpty()) {
                    System.out.println(0);
                } else {
                    System.out.println(1);
                }
            } else if (str.equals("top")) {
                if (stack.isEmpty()) {
                    System.out.println(-1);
                } else {
                    System.out.println(stack.peek());
                }
            }
        }
    }
}