자바 스택 구현

원래 백준 10828을 풀다가 

스택의 갯수의 동적할당이 아니라 명령어의 수를 동적할당으로 하라고 했는데

잘 못읽어서 만들었다. 런타임 에러 나오는 거 보고 다시 읽어봐서 발견..

그냥 지우긴 아까워서 기록한다.

백준 10828 기반에 스택 갯수 동적할당 코드

import java.util.Scanner;

public class algo {

    public static void main(String[] args) {

        int count = 0;
        int push = 0;
        Scanner sc = new Scanner(System.in);

            int size = sc.nextInt();
            int array[] = new int[size];

        while(true) {

            String value = sc.nextLine();

            if (value.startsWith("push")) {
                String temp = value.replace("push ", "");
                push = Integer.valueOf(temp);
                value = "push";
            }

            switch (value) {
                case "push":
                    array[count] = push;
                    count++;
                    break;
                case "pop":
                    System.out.println(array[count-1]);
                    count--;
                    array[count] = 0;
                    break;
                case "size":
                    System.out.println(count);
                    break;
                case "empty":
                    if (array[0] == 0) {
                        System.out.println("1");
                    } else {
                        System.out.println("0");
                    }
                    break;
                case "top":
                    if (array[count-1] != 0 && count != 0) {
                        System.out.println(array[count-1]);
                    } else {
                        System.out.println("-1");
                    }
                    break;
            }
        }
    }

}