[백준] 1874번 스택수열 - Java

[백준] 1874번 스택수열 - Java

 

문제 출처

 

www.acmicpc.net/problem/1874

1874번: 스택 수열

---

※ 풀이

 

스택 개념에 대한 활용 문제이다.

스택을 직접 구현해도 되지만 자바 패키지에서 기본적으로 제공해주는 Stack 을 활용하여 풀어보았다.

 

 

---

※ 소스코드

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

public class Main {

    public static void main(String[] args) throws IOException {
        //input
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringBuilder sb = new StringBuilder();

        Stack<Integer> stack = new Stack<>();
        int N = Integer.parseInt(br.readLine());
        int start = 0;
        while (N-- > 0) {
            int value = Integer.parseInt(br.readLine());
            if (value > start) {
                for (int i = start + 1; i <= value; i++) {
                    stack.push(i);
                    sb.append('+').append('\n');
                }
                start = value;
            }
            else if (stack.peek() != value) {
                System.out.println("NO");
                return;
            }
            stack.pop();
            sb.append('-').append('\n');
        }
        System.out.println(sb);
    }
}