[백준] 28278번: 스택 2 _ JAVA

아로롱·2024년 12월 8일

algorithm

목록 보기
12/25

📍 문제

https://www.acmicpc.net/problem/28278

답안 코드


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));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        Stack<Integer> stack = new Stack<>();

        int N = Integer.parseInt(br.readLine());
        for (int i = 0; i < N; i++) {
            String line = br.readLine();
            if (line.contains(" ")) {
                StringTokenizer str = new StringTokenizer(line);
                str.nextToken();
                stack.add(Integer.parseInt(str.nextToken()));
            } else {
                int num = Integer.parseInt(line);
                switch (num) {
                    case 2:
                        if (!stack.isEmpty()) {
                            bw.write(String.valueOf(stack.pop()));
                        } else bw.write("-1");
                        break;
                    case 3:
                        bw.write(String.valueOf(stack.size()));
                        break;
                    case 4:
                        if (!stack.isEmpty()) {
                            bw.write("0");
                        } else bw.write("1");
                        break;
                    case 5:
                        if (!stack.isEmpty()) {
                            bw.write(String.valueOf(stack.peek()));
                        } else bw.write("-1");
                        break;
                }
                bw.newLine();
            }
        }
        bw.flush();
        bw.close();
        br.close();
    }
}

💡

Buffered 스터디 후 활용하면서 시간 초과 문제를 해결,
약 6개월 전 실패했던 문제에 재도전 성공하였다 ^ ^ V !

profile
Dilige, et fac quod vis

0개의 댓글