백준) 도키도키 간식드리미 (12789번)

12789번: 도키도키 간식드리미

import java.io.*;
import java.util.*;

public class Main {

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Stack<Integer> stack = new Stack<>();

        int n = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine());

        int count = 1;

        for (int i = 0; i < n; i++) {
            int currentStudent = Integer.parseInt(st.nextToken());
            if (currentStudent == count) {
                count++;
                while (!stack.isEmpty() && stack.peek() == count) {
                    stack.pop();
                    count++;
                }
            } else {
                stack.push(currentStudent);
            }
        }

        while (!stack.isEmpty()) {
            if (stack.peek() == count) {
                stack.pop();
                count++;
            } else {
                break;
            }
        }

        System.out.println(stack.isEmpty() ? "Nice" : "Sad");


    }

}

깨달은 점:

stack.size로 반복할 땐 항상 최신 stack.size 기준으로 반복문이 돌아감

stack.size로 반복할 땐 stack.size를 변동시키는 연산을 하면 원하지 않는 결과가 나올 수 있음.