[B0J]스택(10828번)

티어 : Sliver 4
정답여부 : 정답
알고리즘 유형 : 구현, 자료구조,스택
시간 제한 : 0.5초

 

 

💡문제

● 스택

 

 

💡입력1

14
push 1
push 2
top
size
empty
pop
pop
pop
size
empty
pop
push 3
empty
top

 

 

💡출력1

2
2
0
2
1
-1
0
1
-1
0
3

 

💡입력2

7
pop
top
push 123
top
pop
top
pop

 

 

💡출력2

-1
-1
123
123
-1
-1

 

 

💡문제 분석

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

각 스택 명령어로 이 말 그대로 코드를 구현하면 되는 스택의 기본 문제 

 

 

💡알고리즘 설계 

1. 각 명령어 수를 입력받음

2. 스택을 구현하고 명령어 만큼 입력값을 줌

3. 각 조건에 따라 값을 구하면 되는 문제

 

💡시간복잡도

• O(N)

 

💡코드

import java.io.*;
import java.util.Stack;
import java.util.StringTokenizer;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		Stack<Integer> stack  = new Stack<Integer>();
		for (int i = 0; i < N; i++) {
			StringTokenizer st = new StringTokenizer(br.readLine()); 
			String s = st.nextToken();
			
			if(s.equals("push")) {
				int c = Integer.parseInt(st.nextToken());
				stack.push(c);
			}
			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("top")) {
				if(stack.isEmpty()) {
					System.out.println(-1);
				}
				else {
					System.out.println(stack.peek());
				}
			}
			else if (s.equals("empty")) {
				if(stack.isEmpty()) {
					System.out.println(1);
				}
				else {
					System.out.println(0);
				}
			}
			
        }
	}
}

 

💡 느낀점 or 기억할정보

스택의 기본 문제라서 조금 기본기 다지는거 같아 괜찮은거 같다.