배열과 리스트: 스택 수열 (백준 1874)

 

1874번: 스택 수열

 

 

 

 

난이도: 실버 3

풀이: Java가 제공하는 유틸 중 하나인 Stack을 사용하여 푸는 문제이다. 그리고 Stack이 pop, push 작업을 반복할 때마다 StringBuffer에 저장할 수 있도록 한다.

 

import java.util.*;
 
public class Q1874 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N     = scanner.nextInt();
        int A[]    = new int[N];
        for (int i = 0; i < N; i ++) {
            A[i] = scanner.nextInt();
        }
        
        Stack<Integer> stack = new Stack<>();
        StringBuffer bf = new StringBuffer();
        int num = 1;
        boolean result = true;
        
        for (int i = 0; i < A.length; i ++) {
            int S = A[i];
            if (S >= num){
                while (S >= num) {
                    stack.push(num ++);
                    bf.append("+\n");
                }
                stack.pop();
                bf.append("-\n");
            } else {
                int n = stack.pop();
                if (n > S) {
                    System.out.println("NO");
                    result = false;
                    break;
                } else {
                    bf.append("-\n");
                }
            }
        }
        
        if (result) {
            System.out.println(bf.toString());
        }
    }
}