10828 스택

스택

시간 제한메모리 제한제출정답맞힌 사람정답 비율

 

0.5 초 (추가 시간 없음)

256 MB

297634

112494

81424

38.487%

문제

정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.

명령은 총 다섯 가지이다.

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

입력

첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.

출력

출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다. 

 

나의 답안 (성공, 약 16분 소요)

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.io.IOException;
import java.util.Stack;
import java.util.StringTokenizer;

/*
* 1. 첫째 줄에는 명령의 수가 주어진다.
* 2. 먼저 문자열을 공백 기준으로 나누고, 처음 문자로 push, pop, size, empty 중 같은 문자열을 찾는다.
* 3. 해당 문자열과 같은 이름을 가진 메서드를 호출하여 수행
* - 단, push 는 스택에 넣은 이후, return 값이 없다.
* - pop 은 가장 위에 있는 정수를 빼고, 그 수를 출력, 정수가 없으면 -1
* - size 는 스택에 들어있는 정수 개수 출력
* - empty 는 스택이 비어있으면 1, 아니면 0을 출력
* - top 는 스택의 가장 위에 있는 정수를 출력한다. 만약 스탹에 들어있는 정수가 없는 경우 -1 출력*/

public class Main {
    public static void main(String[] args) throws IOException {
        // 입력값 읽을 준비
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        // 스택 생성
        Stack<String> stack = new Stack<>();

        // 테스트 케이스 개수
        int T = Integer.parseInt(br.readLine());

        // 개수만큼 반복
        for(int i = 0; i < T; i++){
            StringTokenizer st = new StringTokenizer(br.readLine(), " ");
            String s = st.nextToken();

            // push일 경우
            if(s.equals("push")){
                stack.push(st.nextToken());
            } else if(s.equals("pop")) {
                if(stack.isEmpty()){
                    System.out.println(-1);
                } else {
                    System.out.println(stack.pop());
                }
            } else if(s.equals("size")){
                System.out.println(stack.size());
            } else if(s.equals("empty")){
                if(stack.isEmpty()){
                    System.out.println(1);
                } else {
                    System.out.println(0);
                }
            } else if(s.equals("top")){
                if(stack.isEmpty()){
                    System.out.println(-1);
                } else {
                    System.out.println(stack.peek());
                }
            }
        }
    }
}

 

 💡회고

• if문 보다는 switch문으로 깔끔하게 적는 코드가 많이 보였다.
• pop, push 등을 메서드로 따로 빼서, 메서드 호출로 작성하는 코드도 많이 보였다.
• System.out.println() 말고 stream을 사용해서 출력하는 방법도 해봐야겠다.
• 코드가 깔끔해보이지 않아서 그 부분을 조금.... 신경써야 할 것 같다.