[JAVA] 프로그래머스 LEVEL2 택배상자

https://school.programmers.co.kr/learn/courses/30/lessons/131704

프로그래머스

코딩테스트 연습 > 연습문제 > 택배상자

 

난이도: LEVEL2

 

정답 코드

import java.util.*;
import java.util.stream.*;
class Solution {
    public int solution(int[] order) {
         int answer = 0;
         Stack<Integer> stack = new Stack<>();
         int cur = 1; // 어차피 컨테이너 벨트에는 1 ~ n 순서로 들어있음 
         int cnt = 0;        
         while(cnt < order.length){
             int target = order[cnt++];
             if(cur <= target){
                 while(cur != target){
                     stack.push(cur++);
                 }
                 cur++;
                 answer++;
             }
             else{ // cur > target 이라면 이미 컨테이너 벨트에서 보조 컨테이너 벨트로 이동 상태 
                if(stack.isEmpty() || stack.peek() != target) break;
                else{
                    answer++;
                    stack.pop();
                } 
             }
         }
         return answer;
    }
}