백준 10828 스택 | 자바 java | BufferedReader BufferedWriter

 

이번 포스팅은 백준 10828번 스택입니다.

아래 url를 클릭하시면 백준 사이트에서 문제를 볼 수 있습니다.

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

10828번: 스택

 

 

 

기본 알고리즘

• 구현
• 자료구조
• 스택

 

반응형

풀이

스택 자료구조를 배열을 사용해서 구현함.

Scanner와 System.out.println() 사용시 시간 초과 발생함.

BufferedReader, BufferedWriter 사용함.

//BufferedWriter로 int형 변수 출력
bw.write(String.valueOf(999));

 

 

전체 코드

 

import java.io.*;
import java.util.StringTokenizer;
 
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));
 
        String num = br.readLine();
        int n = 0;
        n = Integer.parseInt(num);
 
        int stack[] = new int[n];
        int size=0;
 
        String order = "";
        StringTokenizer st;
        String output="";
        for(int i=0; i<n; i++){
            order = br.readLine();
            st = new StringTokenizer(order);
            output="";
 
            switch (st.nextToken()) {
                case "push":
                    stack[size]=Integer.parseInt(st.nextToken());
                    size++;
                    break;
                case "pop":
                    if(size==0){
                        output = String.valueOf(-1);
                    } else {
                        output = String.valueOf(stack[size-1]);
                        size--;
                    }
                    break;
                case "size":
                    output = String.valueOf(size);
                    break;
                case "empty":
                    if(size==0){
                        output=String.valueOf(1);
                    } else{
                        output=String.valueOf(0);
                    }
                    break;
                case "top":
                    if(size==0){
                        output = String.valueOf(-1);
                    } else {
                        output = String.valueOf(stack[size-1]);
                    }
                    break;
                default:
                    break;
            }
 
            if(output.length()>0){
                bw.write(output);
                bw.newLine();
                bw.flush();
            }
 
        }
        
        bw.flush();
        bw.close();
    }
}