[JAVA-자바] 10828번: 스택

 

 

소스코드:

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Stack;
import java.util.StringTokenizer;

public class Main {
    static BufferedReader br;
    static BufferedWriter bw;

    static int N;

    static Stack<Integer> stack;

    public static void main(String[] args) throws IOException {
        br = new BufferedReader(new InputStreamReader(System.in));
        bw = new BufferedWriter(new OutputStreamWriter(System.out));

        N = Integer.parseInt(br.readLine());
        stack = new Stack<>();

        for (int i = 0; i < N; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            String command = st.nextToken();
            switch (command) {
                case "push" -> {
                    stack.push(Integer.parseInt(st.nextToken()));
                }
                case "pop" -> {
                    bw.write((stack.empty() ? -1 : stack.pop()) + "\n");
                }
                case "size" -> {
                    bw.write(stack.size() + "\n");
                }
                case "empty" -> {
                    bw.write((stack.empty() ? 1 : 0) + "\n");
                }
                case "top" -> {
                    bw.write((stack.empty() ? -1 : stack.lastElement()) + "\n");
                }
            }
        }
        br.close();

        bw.flush();
        bw.close();
    }
}

 

글의 내용 중 잘못된 점이나 수정이 필요한 부분, 혹은 궁금한 사항이 있다면 언제든 댓글로 남겨주시면 감사하겠습니다.

여러분의 피드백은 더 나은 글을 작성하는 데 큰 도움이 됩니다. 감사합니다.