[java 백준] 실버 3/1874번 스택수열

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

1874번: 스택 수열

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Stack;
 
public class Main {
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Stack<Integer> stack = new Stack<>();
        ArrayList<String> arr = new ArrayList<>();
       
        int num = Integer.parseInt(br.readLine());
        int result = 0;
        stack.push(result);
        for (int i = 1; i <= num; i++) {
            int n = Integer.parseInt(br.readLine());
            while (stack.peek() < n) {
                result++;
                stack.push(result);
                arr.add("+");
 
                if (stack.peek() == n) {
                    break;
                }
 
            }
            if (stack.peek() == n) {
                arr.add("-");
                stack.pop();
            }
            if (stack.peek() > n) {
                arr.clear();
                arr.add("NO");
                break;
            }
 
        }
        for (int i = 0; i < arr.size(); i++) {
            System.out.println(arr.get(i));
        }
 
    }
 
}

stack.peek()와 4가 같기 때문에 pop을 해주면서 arr에 -가 추가되고, stack에는 숫자가 0,1,2,3이 남는다.

 

이런 규칙으로 stack과 arr이 만들어진다. 7부터 1까지는 어차피 다 - 이므로 생략하겠다.