본문 바로가기

컴퓨터/백준 알고리즘

백준 알고리즘 - 28278번: 스택 2 [Java]

문제: https://www.acmicpc.net/problem/28278

 

28278번: 스택 2

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

www.acmicpc.net

1. 코드

import java.util.*;
public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int N = sc.nextInt();
		Stack <Integer> stack = new Stack<>();
		
		StringBuilder sb = new StringBuilder();
		while(N-- > 0) {
			int cmd = sc.nextInt();
			switch(cmd) {
			case 1: stack.push(sc.nextInt()); break;
			case 2:
				if(stack.isEmpty())
					sb.append(-1).append('\n');
				else
					sb.append(stack.pop()).append('\n');
				break;
			case 3: sb.append(stack.size()).append('\n'); break;
			case 4:
				if(stack.isEmpty())
					sb.append(1).append('\n');
				else
					sb.append(0).append('\n');
			break;
			case 5:
				if(stack.isEmpty())
					sb.append(-1).append('\n');
				else
					sb.append(stack.peek()).append('\n');
			break;
			}
		}
		System.out.println(sb.toString());
	}
}

2. 설명

StringBuilder를 이용하여 문자열을 한번에 출력하는 내용외에는 특별한 것이 없습니다.