[JAVA] 백준 10828번 【스택】

---

백엔드 스쿨에서 한번 했었던, 스택 구현하기 문제였다.

 

스택이 어떤 데이터 구조인지 알면, 더 쉽게 풀 수 있지만, 몰라도 위 설명을 보고 위 기능을 구현하는 매서드를 만들면 충분히 해결할 수 있다.

 

먼저, 명령어 수가 10000개 주어지므로, 모든 명령어가 push인 경우에도 데이터 공간은 10000개면 충분하기 때문에, 클래스 변수로 index = -1, 10001 크기의 int 배열을 생성하였다.

 

그리고 5개의 static 메서드를 구현하였다.

 

push 의 경우 데이터를 저장만하면 되므로, index를 전위연산자로 1 증가시켜준 후 배열에 입력하였다.

 

pop은 index가 -1을 가리키고 있으면 스택이 비어있다는 의미이므로 조건식을 활용해 index가 -1인 경우 -1을 출력하게 하였고, 그 외는, 해당 데이터를 출력하고 index 를 후위연산자로 1 감소시켰다.

 

pop 메서드에서, 굳이 데이터를 출력하고 0으로 만들어주지 않아도, 인덱스 값을 통해서 다른 기능을 다 구현할 수 있다!

 

size 메서드는, index값에 +1 을 해주면 되는 간단한 메서드이고

 

empty와 top 메서드 역시 index값이 -1 인 경우와 아닌 경우로 나누어 기능을 구현하면 된다!

 

 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

public class Main {
    static int index = -1;
    static int[] arr = new int[10001];

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.valueOf(br.readLine());
        for (int i = 0; i < N; i++) {
            String[] input = br.readLine().split(" ");
            if (input[0].equals("push")) {
                push(Integer.valueOf(input[1]));
            } else if (input[0].equals("top")) {
                top();
            } else if (input[0].equals("size")) {
                size();
            } else if (input[0].equals("empty")) {
                empty();
            } else if (input[0].equals("pop")) {
                pop();
            }
        }

    }


    static void push(int input) {
        arr[++index] = input;
    }

    static void pop() {
        if (index == -1) {
            System.out.println(-1);
        } else {
            System.out.println(arr[index--]);
        }
    }

    static void size() {
        System.out.println(index + 1);
    }

    static void empty() {
        if (index == -1) {
            System.out.println(1);
        } else {
            System.out.println(0);

        }
    }

    static void top() {
        if (index == -1) {
            System.out.println(-1);
        } else {
            System.out.println(arr[index]);

        }
    }

}