[백준/JAVA] 10828 스택

겨우 성공...

 

 

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

10828번: 스택

 

단순 스택 구현 문제라 간단할줄 알았는데...

자꾸 시간초과 / 런타임에러가 떴다

 

<처음에 내가 구현했던 코드>

import java.util.Stack;

public class bj_10828 {
	
	public static void main(String[] args) {
		
    Scanner sc = new Scanner(System.in);

    int N = sc.nextInt();

    Stack<Integer> stack = new Stack<>();
		
		
    for(int i=0; i<N; i++) {

        String od = sc.next();

        if(od.contains("push")) {
            int X = sc.nextInt();
            stack.push(X); //stack에 X값 push
        }else if(od.equals("pop")) {
            System.out.println(stack.isEmpty()?-1:stack.pop()); //stack이 비어있으면 -1출력, 아니면 pop하고 출력
        }else if(od.equals("size")) {
            System.out.println(stack.size()); 
        }else if(od.equals("empty")) {
            System.out.println(stack.empty());
        }else if(od.equals("top")) {
            System.out.println(stack.isEmpty()?-1:stack.peek()); //stack이 비어있으면 -1출력, 아니면 상단값 출력
        }
      }
   }
}

 

이클립스에서는 제대로 돌아갔지만 제출하니까 시간초과가 떴다

그래서 Scanner를 BufferedReader로 수정하고 다시 제출하니 런타임에러

첫번째 if문에서 br.readLine()으로 읽기만 하고 공백처리가 안된게 원인이였던 것 같다.

 

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 NumberFormatException, IOException {
        
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
      	
        Stack<Integer> stack = new Stack<Integer>();
		
        int N = Integer.parseInt(br.readLine());
				
		for(int i=0; i<N; i++) {

            String[] od = br.readLine().split(" ");
			
			switch(od[0]){
                case "push":
                    stack.push(Integer.parseInt(od[1]));
                    break;
                case "pop":
                    if(stack.empty()){
                        bw.write("-1" + "\n");
                    }else{
                        bw.write(stack.pop() + "\n");
                    }
                    break;
                case "size":
                    bw.write(stack.size() + "\n");
                    break;
                case "empty":
                    if(stack.empty()){
                        bw.write("1"+"\n");
                    }else{
                        bw.write("0"+"\n");
                    }
                    break;
                case "top":
                    if(stack.empty()){
                        bw.write("-1" + "\n");
                    }else{
                        bw.write(stack.peek() + "\n");
                    }
                    break;
            }
		}
        bw.flush();
        bw.close();
	}
}

 

switch문이 더 깔끔해보여서 수정했다.

그리고 System.out.print 대신 BufferedWriter로 수정

메모리 사용에 대한 고민을 해봐야 할 것 같다.