[백준] 10828번 - 스택 [Java]

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

 

[백준] 10828번 - 스택 [Java]

[백준] 10828번 - 스택 [Java]

---

1.  아이디어

 

Stack을 직접 구현하는 문제로 편의상 Deque를 활용했다.

---

2. 문제풀이

 

명령의 처리는 switch문을 활용해서 구현했다.

---

3. 코드

 

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

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

        Deque<Integer> stack = new ArrayDeque<>();

        int N = Integer.parseInt(br.readLine());

        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());

            String order = st.nextToken();

            switch (order) {
                case "push":
                    int X = Integer.parseInt(st.nextToken());
                    stack.push(X);
                    break;

                case "pop":
                    if (stack.isEmpty()) sb.append(-1).append("\n");
                    else sb.append(stack.pop()).append("\n");
                    break;

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

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

                case "top":
                    if (stack.isEmpty()) sb.append(-1).append("\n");
                    else sb.append(stack.peek()).append("\n");
                    break;
            }
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

---

4. 후기