백준 1874 스택 수열(Java)

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

1874번: 스택 수열

---

 

해석 및 팁

 

---

 

이 문제는 문제를 이해하기만 하면 쉽게 풀 수 있습니다. 예제처럼 4가 입력되려면 스택 안에 1,2,3,4 가 있어야 하므로 "+"가 4개 필요하다. 그다음 4를 pop 해준 뒤 3도 pop 해주면 된다 6이 되려면 스택에 6이 필요하므로 순서대로 5와 6을 push 해줍니다. 그다음 8이 필요하므로 순서대로 7과 8을 push 해줍니다. 즉 오름차순으로 계산하면 되는 문제입니다.

 

---

 

Java 코드

 

---

import java.util.Scanner;
import java.util.Stack;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    StringBuilder sb = new StringBuilder();
    Stack<Integer> stack = new Stack<>();

    int n = sc.nextInt();
    int count = 1;
    
    for(int i = 1; i <= n; i++) {
      int num = sc.nextInt();

      while(num >= count) {
        stack.push(count);
        sb.append("+").append("\n");
        count++;
      }

      if(stack.peek() == num) {
        stack.pop();
        sb.append("-").append("\n");
      }
        
      else {
        sb.setLength(0);    //StringBuidler 초기화
        sb.append("NO");
        break;
      }
     
    }
    
    System.out.println(sb);
  }
}