Categories: 백준

백준 10828 스택 (자바 Java)

java

백준 10828 스택

문제 정보

문제 링크

문제 출처 : 백준 Baekjoon
문제 번호 : 10828
문제 제목 : 스택
문제 난이도 : 실버4
풀이 언어 : Java

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 문제

push X: 정수 X를 스택에 넣는 연산이다.
pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
size: 스택에 들어있는 정수의 개수를 출력한다.
empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.

접근

Scanner와 System.out.println은 시간 초과가 뜨기 때문에 BufferedReader와 BufferedWrite를 사용해 문제를 풀었다.

풀이

BufferedReader 를 사용하여 입력을 받는다.
BufferedWriter 를 사용하여 결과를 출력한다.
명령어의 총 수를 입력 받아 정수로 변환한다.
정수를 저장할 수 있는 스택을 초기화한다.
입력 받은 명령어 수만큼 반복하여 각 명령어를 처리한다.
입력된 명령어를 공백으로 구분하여 배열로 변환한 후, 해당 명령어에 맞는 작업을 수행한다.
- push 명령어 처리: 스택에 새로운 요소를 추가한다.
- pop 명령어 처리: 스택에서 가장 위에 있는 요소를 제거하고 그 값을 출력한다.
- size 명령어 처리: 스택에 저장된 요소의 수를 출력한다.
- empty 명령어 처리: 스택이 비어있는지 확인하고 그 결과를 출력한다.
- top 명령어 처리: 스택의 가장 위에 있는 요소를 출력한다.

후기

스택을 활용한 기본적인 문제이다. 스택과 관련된 명령어들을 잘 숙지해야겠다.

코드

Java

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Stack;

public class Main {
 public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        int N = Integer.parseInt(br.readLine());
        Stack<Integer> stack = new Stack<>();

        for (int i = 0; i < N; i++) {
            String[] input = br.readLine().split(" ");

            switch (input[0]) {
                case "push":
                    stack.push(Integer.parseInt(input[1]));
                    break;
                case "pop":
                    if (!stack.isEmpty()) {
                        bw.write(stack.pop() + "\n");
                    } else {
                        bw.write(-1 + "\n");
                    }
                    break;
                case "size":
                    bw.write(stack.size() + "\n");
                    break;
                case "empty":
                    bw.write((stack.isEmpty() ? 1 : 0) + "\n");
                    break;
                case "top":
                    bw.write((stack.isEmpty() ? -1 : stack.peek()) + "\n");
                    break;
            }
        }
        
        br.close();
        bw.flush();
        bw.close();
 }
}