[Java] Baekjoon Online Judge 28278: 스택 2

28278번: 스택 2

💡문제

---

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

 

명령은 총 다섯 가지이다.

1. X: 정수 X를 스택에 넣는다. (1 ≤ X ≤ 100,000)
2. 스택에 정수가 있다면 맨 위의 정수를 빼고 출력한다. 없다면 -1을 대신 출력한다.
3. 스택에 들어있는 정수의 개수를 출력한다.
4. 스택이 비어있으면 1, 아니면 0을 출력한다.
5. 스택에 정수가 있다면 맨 위의 정수를 출력한다. 없다면 -1을 대신 출력한다.

 

✏️입력

---

첫째 줄에 명령의 수 N이 주어진다. (1 ≤ N ≤ 1,000,000)
둘째 줄부터 N개 줄에 명령이 하나씩 주어진다.
출력을 요구하는 명령은 하나 이상 주어진다.

 

📑출력

---

출력을 요구하는 명령이 주어질 때마다 명령의 결과를 한 줄에 하나씩 출력한다.

 

예제입력 1

9
4
1 3
1 5
3
2
5
2
2
5

 

예제출력 1

1
2
5
3
3
-1
-1

 

🔎알고리즘 분류

---

• 자료 구조
• 스택

 

📌풀이

---

코드 1

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

public class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); // 입력을 위한 BufferedReader
		StringBuilder sb = new StringBuilder(); // 출력 결과 저장을 위한 StringBuilder 객체
		Stack<Integer> stack = new Stack<Integer>(); // 정수를 저장할 스택
		StringTokenizer st; // StringTokenizer를 사용하여 문자열을 토큰으로 분리하기 위한 객체
		int N = Integer.parseInt(br.readLine()); // 명령의 개수

		for (int i = 0; i < N; i++) {
			st = new StringTokenizer(br.readLine()); // 한 줄씩 명령을 입력 받아서 StringTokenizer를 통해 명령과 값 분리
			String command = st.nextToken();

			if (command.equals("1")) // "1 X": 정수 X를 스택에 넣음
				stack.push(Integer.parseInt(st.nextToken()));
			else if (command.equals("2")) { // "2": 스택에서 맨 위의 정수를 빼고 출력, 스택이 비어있으면 -1 출력
				if (!stack.isEmpty()) {
					sb.append(stack.lastElement()).append("\n");
					stack.pop();
				} else
					sb.append(-1).append("\n");
			} else if (command.equals("3")) // "3": 스택에 들어있는 정수의 개수를 출력
				sb.append(stack.size()).append("\n");
			else if (command.equals("4")) // "4": 스택이 비어있으면 1, 아니면 0을 출력
				if (stack.isEmpty())
					sb.append(1).append("\n");
				else
					sb.append(0).append("\n");
			else if (command.equals("5")) // "5": 스택이 비어있으면 -1을 출력, 아니면 맨 위의 정수 출력
				if (stack.isEmpty())
					sb.append(-1).append("\n");
				else
					sb.append(stack.lastElement()).append("\n");
		}

		br.close();
		System.out.println(sb);
	}
}

 

코드 2

import java.util.Stack;

public class Main {

	public static void main(String[] args) throws Exception {
		Stack<Integer> stack = new Stack<>();
		StringBuilder sb = new StringBuilder();
		int n = read();

		for (int i = 0; i < n; i++) {
			int order = read();

			int x = 0;
			if (order == 1) {
				x = read(); // 명령이 1인 경우 정수를 입력
			}

			switch (order) {
			case 1:
				// 1 X: 정수 X를 스택에 넣음
				stack.push(x);
				break;
			case 2:
				// 2: 스택에서 맨 위의 정수를 빼고 출력, 스택이 비어있으면 -1 출력
				sb.append(!stack.isEmpty() ? stack.pop() : -1).append('\n');
				break;
			case 3:
				// 3: 스택에 들어있는 정수의 개수를 출력
				sb.append(stack.size()).append('\n');
				break;
			case 4:
				// 4: 스택이 비어있으면 1, 아니면 0을 출력
				sb.append(stack.isEmpty() ? 1 : 0).append('\n');
				break;
			case 5:
				// 5: 스택이 비어있으면 -1을 출력, 아니면 맨 위의 정수 출력
				sb.append(!stack.isEmpty() ? stack.peek() : -1).append('\n');
				break;
			}
		}

		System.out.println(sb);
	}

	private static int read() throws Exception {
		int c, n = System.in.read() & 15; // 초기값

		while ((c = System.in.read()) > 32)
			n = (n << 3) + (n << 1) + (c & 15); // 숫자를 읽어와서 정수로 변환
		if (c == 13)
			System.in.read(); // 개행 문자('\r')가 나오면 다음 문자를 읽음

		return n; // 읽은 정수 반환
	}

}