스택구현하기

스택 구현하기

 

문제

---

이 문제에서는 스택을 구현한다. 스택은 다음 세 개의 연산을 지원한다. Push X : 스택에 정수 X를 push한다. 만약 스택이 꽉 차서 push를 할 수 없다면, “Overflow”를 출력한다. Pop : 스택에서 정수 하나를 pop한다. 만약 스택이 비어있어서 pop을 할 수 없다면, “Underflow”를 출력한다. Top : 스택의 top에 있는 정수를 출력한다. 만약 스택이 비어있다면 “NULL”을 출력한다. 크기가 n인 스택에 m개의 연산을 하는 프로그램을 작성하시오. 입력의 편의를 위해서 Push는 “1”, Pop은 “2”, Top은 “3”으로 표현한다. 다음은 크기 4인 스택에 10개의 연산이 입력으로 주어졌을 때의 입력과 출력의 예이다.

XYZ

 

입력

---

첫째 줄에 스택의 크기 n, 연산의 개수 m이 주어진다. ( 1 <= n <= 100, 1 <= m <= 1,000 ) 두 번째 줄부터 연산이 주어진다. 1은 Push, 2는 Pop, 3은 Top 연산을 의미한다.  

출력

---

연산의 결과를 출력한다.

 

예제 입력

4 10
1 1
1 2
1 3
2
3
1 4
1 5
3
1 6
3

예제 출력

2
5
Overflow
5

코드

//스택구현

import java.util.*;

public class Main {

    public static ArrayList<Integer> box = new ArrayList<>();
    public static int[] stack;
    public static int stackCount = 0;

    //첫째 줄에 스택의 크기 n, 연산의 개수 m이 주어진다. ( 1 <= n <= 100, 1 <= m <= 1,000 )
    // 두 번째 줄부터 연산이 주어진다. 1은 Push, 2는 Pop, 3은 Top 연산을 의미한다.
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        //총 몇개의 숫자를 입력 받을 것인지
        int a;
        int b;
        int c;
        int d = 0;


        a = sc.nextInt();
        stack = new int[a];

        b = sc.nextInt();

        for (int i = 0; i < b; i++) {
            c = sc.nextInt();
            if (c == 1) {
                d = sc.nextInt();
            }
            operation(c, d);
        }

        sc.close();


    }


    public static void operation(int op, int data) {
        if (op == 1) {
            if (stackCount > stack.length - 1) System.out.println("Overflow");
            else {
                stack[stackCount] = data;
                stackCount++;
            }
        } else if (op == 2) {
            if (stackCount <= 0) System.out.println("Underflow");
            else stackCount--;

        } else if (op == 3) {
            if (stackCount <= 0) System.out.println("NULL");
            else System.out.println(stack[stackCount - 1]);
        }
    }


}