[10828.java] 스택

seungyeon·2023년 9월 12일

스택(Stack)
마지막에 저장한 데이터를 가장 먼저 꺼내는 구조 LIFO
: ArrayList와 같은 배열기반의 컬렉션이 적합

알아두기

empty() //Stack이 비었는지
peek() //Stack의 맨 위에 저장된 객체를 반환, 객체를 꺼낸 것은 아님
pop() //Stack의 맨 위에 저장된 객체를 꺼냄
push(item) //Stack에 객체(item)를 저장
int search(Object o) //Stack에서 주어진 객체(o)를 찾아서 그 위치를 반환. 못 찾으면 -1을 반환(index는 1부터 시작)

풀이코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();
        Stack<Integer> st = new Stack<Integer>();

        int N = Integer.parseInt(br.readLine());
        for (int i = 0; i < N; i++) {
            StringTokenizer str = new StringTokenizer(br.readLine(), " ");
            String S = str.nextToken();

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

                case "pop":
                    if(st.isEmpty()){
                        sb.append("-1\n");
                    }else{
                        sb.append(st.pop()+"\n");
                    }
                    break;
                case "size":
                    sb.append(st.size() +"\n");
                    break;
                case "empty":
                    if(st.empty()){
                        sb.append("1\n");
                    }else{
                        sb.append("0\n");
                    }
                    break;
                case "top":
                    if(st.isEmpty()){
                        sb.append("-1\n");
                    }else{
                        sb.append(st.peek()+"\n");
                    }
                    break;
            }
        }
        System.out.println(sb);
    }
}

0개의 댓글