[백준_1874] 스택 수열 - JAVA

문제 출처

https://www.acmicpc.net/problem/1874

풀이

• 스택 문제이다.
• 입력 받은 값이 top 보다 클 경우 스택에 Push를 해준다. 이 때 '+'는 StringBuilder를 통해 append 해준다.
• 만약 스택의 top이 입력 받은 값과 다르다면 구현할 수 없는 수열이다. "NO"를 return 한다.
• 입력 받은 값이 top 보다 작거나 같을 경우는 Pop을 해주면 된다. 이 때 '-'는 StringBuilder를 통해 append 한다.

코드

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

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer stringTokenizer = new StringTokenizer(bufferedReader.readLine());

        Stack<Integer> stack = new Stack<>();
        StringBuilder stringBuilder = new StringBuilder();
        int top = 0;
        int N = Integer.parseInt(stringTokenizer.nextToken());
        for (int i = 0; i < N; i++) {
            stringTokenizer = new StringTokenizer(bufferedReader.readLine());
            int value = Integer.parseInt(stringTokenizer.nextToken());
            if (value > top) {
                for (int j = top + 1; j <= value; j++) {
                    stack.push(j);
                    stringBuilder.append("+").append("\n");
                }
                top = value;
            } else if (stack.peek() != value) {
                System.out.println("NO");
                return;
            }
            stack.pop();
            stringBuilder.append("-").append("\n");
        }
        System.out.println(stringBuilder);
    }
}

채점 결과