[BOJ] 1874 스택수열 - JAVA

---

1. 풀이

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

public class BOJ_1874_스택수열 {
    public static void main(String[] args) throws IOException {
        BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
        int n = Integer.parseInt(in.readLine());

        Stack<Integer> stack = new Stack<>();
        StringBuilder sb = new StringBuilder();

        int count = 1;

        for (int i = 0; i < n; i++) {
            int num = Integer.parseInt(in.readLine());
            while (true) {
                if (num >= count) {
                    stack.push(count++);
                    sb.append("+\n");
                }

                if (stack.empty() || num < stack.peek()) {
                    System.out.println("NO");
                    return;
                } else if (num == stack.peek()) {
                    stack.pop();
                    sb.append("-\n");
                    break;
                }
            }
        }
        System.out.print(sb);

    }
}

• count : 스택에 넣을 숫자
• num : 입력받은 숫자.

num>=count라면 아직 스택에 있는 수가 모자라므로 push(count++)한다.

위의 조건문을 통과한 뒤에도 스택이 비어있거나 num<stack.peek()라면 더 이상 스택에서 pop 시킬 수 없다. 이 문제는 수열 하나만 판단하기 때문에 return시킨다.

  - ex. num==4이고 stack.peek()==5라면, 4를 pop시키기 위해 5를 먼저 pop 시켜야 하기 때문에 불가능하다.

num==stack.peek()이라면 pop시키고, 다음 num을 입력받는다.

---

2. 결과