[BaekJoon-10828] 스택

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

push X: 정수 X를 스택에 넣는 연산이다.
pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
size: 스택에 들어있는 정수의 개수를 출력한다.
empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

입력

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

출력

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

예제 입력 1

14  
push 1  
push 2  
top  
size  
empty  
pop  
pop  
pop  
size  
empty  
pop  
push 3  
empty  
top

예제 출력 1

예제 입력 2

7
pop
top
push 123
top
pop
top
pop

예제 출력 2

-1
-1
123
123
-1
-1

배열의 끝을 스택의 꼭대기로 사용하기

최대 명령 수가 10,000개이므로 그보다 더 많은 정수를 동시에 쌓을 수 없다. 길이 N의 배열과 현재 원소 수 size만 있으면 된다. push X는 stack[size]에 쓰고 size를 늘린다. pop은 비어 있지 않을 때 size를 줄이며 직전 맨 위 값을 출력한다. 아래 코드는 size를 원소 개수로 정의하므로 빈 스택은 size == 0이다.

예제 2의 앞부분을 추적하면 빈 상태에서 pop과 top은 모두 -1, push 123 뒤에는 size=1과 top=123, 다음 pop 뒤에는 다시 size=0이다. top은 읽기만 하고 size를 바꾸지 않는 점이 pop과 다르다. 출력을 명령마다 flush하면 불필요한 I/O가 반복되므로 결과를 모아서 마지막에 한 번 출력한다.

각 명령은 배열 한 칸을 읽거나 쓰므로 시간 O(N), 입력과 별개인 스택 공간 O(N)이다. pop과 top을 빈 상태에서 호출하는 사례가 핵심 경계 조건이다.

코드


import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;

public class Main {

    public static void main(String[] args) throws Exception {
        stack();
    }
    public static void stack() throws Exception {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter out = new BufferedWriter(new OutputStreamWriter(System.out));
        int n = Integer.parseInt(in.readLine());
        int[] stack = new int[n];
        int size = 0;

        for (int i = 0; i < n; i++) {
            String input = in.readLine();
            if (input.startsWith("push ")) {
                stack[size++] = Integer.parseInt(input.substring(5));
            } else if (input.equals("pop")) {
                out.write(size == 0 ? "-1\n" : stack[--size] + "\n");
            } else if (input.equals("size")) {
                out.write(size + "\n");
            } else if (input.equals("empty")) {
                out.write((size == 0 ? 1 : 0) + "\n");
            } else if (input.equals("top")) {
                out.write(size == 0 ? "-1\n" : stack[size - 1] + "\n");
            }
        }
        out.flush();
        in.close();
        out.close();
    }

}