정수를 저장하는 스택을 구현한 다음, 입력으로 주어지는 명령을 처리하는 프로그램을 작성하시오.
명령은 총 다섯 가지이다.
첫째 줄에 주어지는 명령의 수 N (1 ≤ N ≤ 10,000)
이 주어진다. 둘째 줄부터 N
개의 줄에는 명령이 하나씩 주어진다. 주어지는 정수는 1보다 크거나 같고, 100,000보다 작거나 같다. 문제에 나와있지 않은 명령이 주어지는 경우는 없다.
출력해야하는 명령이 주어질 때마다, 한 줄에 하나씩 출력한다.
14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top
2
2
0
2
1
-1
0
1
-1
0
3
7
pop
top
push 123
top
pop
top
pop
-1
-1
123
123
-1
-1
-문제를 만든 사람: baekjoon
-데이터를 추가한 사람: deunlee, djm03178
import java.io.*;
import java.util.Stack;
public class Code10828 {
public static void main(String[] args) throws IOException {
BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
int N=Integer.valueOf(br.readLine());
BufferedWriter bw=new BufferedWriter(new OutputStreamWriter(System.out));
Stack<Integer> stack=new Stack<>();
for(int i=0;i<N;i++){
String[] temp=br.readLine().split(" ");
if(temp[0].equals("push")){
stack.push(Integer.valueOf(temp[1]));
}
else if(temp[0].equals("top")){
if(stack.empty()){
bw.write(String.valueOf(-1)+"\n");
}
else{
bw.write(String.valueOf(stack.peek())+"\n");
}
}
else if(temp[0].equals("size")){
bw.write(String.valueOf(stack.size())+"\n");
}
else if(temp[0].equals("empty")){
if(stack.empty()){
bw.write(String.valueOf(1)+"\n");
}
else{
bw.write(String.valueOf(0)+"\n");
}
} else if (temp[0].equals("pop")) {
if(stack.empty()){
bw.write(String.valueOf(-1)+"\n");
}
else{
bw.write(String.valueOf(stack.pop())+"\n");
}
}
}
br.close();
bw.flush();
bw.close();
}
}
Stack
클래스를 사용하였다.
stack.peek()
이 top()
과 같은 것 이다.
ArrayList를 쓸까 하다가 시간초과가 날까봐 이번에 새롭게 Stack
클래스로 문제를 풀어보았다.