BAEKJOON 10828번: 스택

Kim Hyen Su·2023년 10월 1일

⏲️ 알고리즘

목록 보기

36/95

🎇 문제 URL

10828번: 스택

🎇 문제 설명

stack 클래스 객체를 만들어 내부 메서드를 사용하는 간단한 예제이다.

🎇 Stack<E> 클래스

Java SE 17 & JDK 17 버전 Doc 참고한 스택 관련 메서드.

boolean empty() : 스택이 비어있는지 여부를 확인하는 메서드.
E peek() : 스택 최상위 요소를 삭제 없이 읽어오는 메서드.
E pop() : 스택 최상위 요소를 삭제 후 읽어오는 메서드.
E push(E item) : 스택 최상위에 새로운 요소를 추가하는 메서드.
int search(Object o) : 스택 내부에 요소를 찾은 뒤 위치를 반환해주는 메서드. 최상부에 위치한 요소는 1을 반환하고 1부터 차례로 커진다. 만약, 찾는 객체가 요소로 존재하지 않는 경우 '-1'을 반환한다.

🎇 제출 코드(내 제출)

import java.io.*;
import java.util.*;

public class Main{
    public static void main(String[] args)throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        StringBuilder sb = new StringBuilder();
        Stack<Integer> stack = new Stack<>();
        
        for(int i=0; i < N; i++){
            String str = br.readLine();
            if(str.contains("push")){
                String tmp = str.substring(5);
                int X = Integer.parseInt(tmp);
                stack.push(X);
            }else if(str.equals("pop")){
                if(!stack.empty()){
                    Integer num = stack.pop();
                    sb.append(num).append("\n");
                }else{
                    sb.append("-1").append("\n");
                }
            }else if(str.equals("size")){
                sb.append(stack.size()).append("\n");
            }else if(str.equals("empty")){
                int tmp = 0;
                tmp = stack.empty() ? 1 : 0;
                sb.append(tmp).append("\n");
            }else{ // top
                if(!stack.empty()){
                    Integer num = stack.peek();
                    sb.append(num).append("\n");
                }else{
                    sb.append("-1").append("\n");
                }
            }
        }
        System.out.println(sb);
        br.close();
    }
}

🎇 문제 의도

문제 의도를 확인해보니 자바에 구현된 Stack 클래스를 사용하는 것이 아닌 스택을 int 배열로 구현하는 것이 맞다.

🎇 Stack을 직접 구현한 코드

import java.io.*;
import java.util.*;
 
public class Main {
 
	public static int[] stack;
	public static int size = 0;
 
	public static void main(String[] args) throws IOException {
 
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		
		StringTokenizer st;
		
		int N = Integer.parseInt(br.readLine());
 
		stack = new int[N];
		
		while (N-- > 0) {
			st = new StringTokenizer(br.readLine(), " ");
 
			switch (st.nextToken()) {
			
			case "push":
				push(Integer.parseInt(st.nextToken()));
				break;
				
			case "pop":
				sb.append(pop()).append('\n');
				break;
				
			case "size":
				sb.append(size()).append('\n');
				break;
				
			case "empty":
				sb.append(empty()).append('\n');
				break;
				
			case "top":
				sb.append(top()).append('\n');
				break;
			}
 
		}
		System.out.println(sb);
	}
 
	public static void push(int item) {
		stack[size] = item;
		size++;
	}
	
	public static int pop() {
		if(size == 0) {
			return -1;
		}
		else {
			int res = stack[size - 1];
			stack[size - 1] = 0;
			size--;
			return res;
		}
	}
	
	public static int size() {
		return size;
	}
	
	public static int empty() {
		if(size == 0) {
			return 1;
		}
		else {
			return 0;
		}
	}
	
	public static int top() {
		if(size == 0) {
			return -1;
		}
		else {
			return stack[size - 1];
		}
	}
	
}