[백준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
2
0
2
1
-1
0
1
-1
0
3

나의 풀이

이 문제는 그냥 스택을 만들면되는 문제였다. 스택을 처음 접해보는 사람은 어려울수도 있겠지만 어느정도 알고리즘 문제를 접해봤다면 아주 쉽게 풀 수 있는 문제 중 하나이다.

우선 List<Integer> stack = new Arraylist()를 사용하여 스택을 만들어준다. ( 물론 이 문제에서는 N을 알려주기 때문에 최대 크기인 14만큼의 int형 배열을 사용해도 되지만 필자는 크기가 정적인 배열보다는 List가 동적으로 사이즈가 조절되고 size를 구해주는 메서드도 있기때문에 List를 사용하였다. )

그리고나서는 각 명령마다 메서드를 만들어주었다.

• push(int num) : 아규먼트로 스택에 넣을 숫자를 입력받고 List에 add()를 사용하여 맨 뒤에 넣어준다.
• pop() : 스택은 후입선출이므로 최근에 들어온게 pop되어야한다. push에서 맨 뒤에 넣어주었으니 맨 뒤의 인자를 반환함과 동시에 스택에서 remove()를 사용하여 삭제해준다.
• sizeOf() : int형 배열을 사용한다면 for문을 돌려 size를 구할 수 있고, List를 사용할 경우 size()를 사용하면 쉽게 구할 수 있다.
• isEmpty() : size()를 구해서 0이면 true , 0이 아니면 false를 반환해주어도 되지만 간단하게 List의 isEmpty()를 사용했다.
• top() : top는 pop과 비슷하지만 삭제는 되지않는 명령이니까 remove를 하지않고 get()으로 값만 반환해준다.

코드

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

/**
 * 10828번 스택
 */
public class Baekjoon10828 {
    static List<Integer> stack = new ArrayList<>();
    static StringTokenizer st = null;

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

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

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

            String op = st.nextToken();

            if (op.equals("push")) push(Integer.parseInt(st.nextToken()));
            else if (op.equals("pop")) System.out.println(pop());
            else if (op.equals("size")) System.out.println(sizeOf());
            else if (op.equals("empty")) {
                if (isEmpty()) System.out.println(1);
                else System.out.println(0);
            }
            else if (op.equals("top")) System.out.println(top());
            else System.out.println("Error");
        }

    }

    private static void push(int num) {
        stack.add(num);
    }

    private static int pop() {
        if (isEmpty()) return -1;
        else return stack.remove(stack.size() - 1);        // remove()는 삭제가 되는 인덱스의 값을 반환과 동시에 삭제를 한다.
    }

    private static int sizeOf() {
        return stack.size();
    }

    private static Boolean isEmpty() {
        return stack.isEmpty();
    }

    private static int top() {
        if (isEmpty()) return -1;
        else return (stack.get(stack.size()-1));
    }
}