[BackJoon-10828] 스택 문제 정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 다섯 가지이다.
push X: 정수 X를 스택에 넣는 연산이다.
pop: 스택에서 가장 위에 있는 정수를 빼고, 그 수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
size: 스택에 들어있는 정수의 개수를 출력한다.
empty: 스택이 비어있으면 1, 아니면 0을 출력한다.
top: 스택의 가장 위에 있는 정수를 출력한다. 만약 스택에 들어있는 정수가 없는 경우에는 -1을 출력한다.
입력 첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)이 주어진다. 둘째 줄부터 N개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력 출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
예제 입력 1 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 14 push 1 push 2 top size empty pop pop pop size empty pop push 3 empty top
예제 출력 1 1 2 3 4 5 6 7 8 9 10 11 2 2 0 2 1 -1 0 1 -1 0 3
예제 입력 2 1 2 3 4 5 6 7 8 7 pop top push 123 top pop top pop
예제 출력 2
[코드] 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 import java.io.BufferedReader;import java.io.BufferedWriter;import java.io.InputStreamReader;import java.io.OutputStreamWriter;import java.util.Scanner;import java.util.Stack;public class Main { public static void main (String[] args) throws Exception { stack(); } public static void stack () throws Exception { BufferedReader in = new BufferedReader(new InputStreamReader(System.in)); BufferedWriter out = new BufferedWriter(new OutputStreamWriter(System.out)); int n = parseInt(in.readLine()); String stack[] = new String[n]; int size = -1 ; for (int i = 1 ; i <= n; i++) { String input = in.readLine(); if (input.contains("push" )) { stack[++size] = input.split(" " )[1 ]; } else if (input.equals("pop" )) { out.write(size == -1 ? size + "\n" : stack[size--] + "\n" ); } else if (input.equals("size" )) { out.write((size + 1 ) + "\n" ); } else if (input.equals("empty" )) { out.write((size == -1 ? 1 : 0 ) + "\n" ); } else if (input.equals("top" )) { out.write((size == -1 ? size : stack[size]) + "\n" ); } out.flush(); } in.close(); out.close(); } public static int parseInt (String arg) { return Integer.parseInt(arg); } }
문제 사이트 : https://www.acmicpc.net/problem/10828