[백준] 10828 스택 - Java

문제

링크

---

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());

        StringBuilder sb = new StringBuilder();
        MyStack stack = new MyStack();
        int value = 0;

        for (int i = 0; i < N; i++) {
            StringTokenizer st = new StringTokenizer(br.readLine());
            String order = st.nextToken();
            if (order.equals("push")) {
                value = Integer.parseInt(st.nextToken());
            }

            switch (order) {
                case "push":
                    stack.push(value);
                    break;

                case "pop":
                    sb.append(stack.pop()).append("\n");
                    break;

                case "size":
                    sb.append(stack.size()).append("\n");
                    break;

                case "empty":
                    sb.append(stack.empty() ? 1 : 0).append("\n");
                    break;

                case "top":
                    sb.append(stack.top()).append("\n");
                    break;
            }
        }
        System.out.print(sb);
    }

    static class MyStack {
        private List<Integer> list = new ArrayList<>();

        void push(int value) {
            list.add(value);
        }

        int pop() {
            if (list.isEmpty()) return -1;

            return list.remove(list.size() - 1);
        }

        int size() {
            return list.size();
        }

        boolean empty() {
            return list.isEmpty();
        }

        int top() {
            if (list.isEmpty()) return -1;
            return list.get(list.size() - 1);
        }
    }
}

---

리뷰

자료구조 중 하나인 Stack을 직접 구현해보는 문제이다.
사실 이 문제를 풀기전에 몇번 구현해보아서 생각보다 쉽게 접근할 수 있었다.
오래걸렸던 부분은 switch문 case작성하는 부분이랄까 😂