백준 28278 자바 (스택2)

<문제>

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

28278번: 스택 2

<풀이>

접근

문제에서 대놓고 스택을 활용해서 접근하라고 주어졌기 때문에 스택 자료구조를 활용하는 식으로 접근했다.

 

기능 목록

스택  메서드 기능

 

<코드>

import java.util.*;
import java.io.*;


public class Main {

    public static void main(String[] args)throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        Stack<Integer> stack = new Stack<>();
        int n = Integer.parseInt(br.readLine());
        for(int i = 0; i < n; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            int m = Integer.parseInt(st.nextToken());
            if(m == 1) {
                int num = Integer.parseInt(st.nextToken());
                stack.push(num);
            }
            else if(m == 2) {
                if(stack.isEmpty()) {
                    sb.append(-1).append("\n");
                }
                else {
                    sb.append(stack.pop()).append("\n");
                }
            }
            else if(m == 3) {
                sb.append(stack.size()).append("\n");
            }
            else if(m == 4) {
                if(stack.isEmpty()) {
                    sb.append(1).append("\n");
                }
                else {
                    sb.append(0).append("\n");
                }
            }
            else {
                if(stack.isEmpty()) {
                    sb.append(-1).append("\n");
                }
                else {
                    sb.append(stack.peek()).append("\n");
                }
            }
        }
        System.out.println(sb);
    }
}

 

<후기>

위 문제 같은 경우 N의 범위가 1,000,000 이므로, 그만큼 출력을 많이 할 수 있는데 자바 같은 경우 System.out.println()의 성능이 좋은 편이 아니라 StringBuilder를 안 쓰면 시간초과가 나온다. 
앞으로 StringBuilder를 자주 사용하는 식으로 문제를 해결해야겠다.