[코테/JAVA] 연습문제 : 택배상자

2022.12.13

코테 문제풀이 22주차 01번 문제

 

 

문제

  ① 연습문제 : 택배상자

 

  ② 링크

프로그래머스

 

 

접근방법

  생각해낸 방법은 1가지

  ① 스택

 

 

풀이

  ① 접근방법 : ①

  ② 풀이

import java.util.Stack;
 
class Solution {
    /*
     * Date: 2022-12-13
     * Time: 21:24
     */
    public int solution(int[] order) {
        return pickUp(order);
    }
 
    /*
     * 짐고르기
     * 1. Stack을 사용해 가장 최근에 쌓은 짐부터 peek
     * 2. Stack.peek() 메소드를 사용해서 값을 비교
     * 3. 실어야 할 짐이면 쌓은 짐의 갯수 카운트
     * 4. 보조 컨테이너 벨트에 올리는 경우 Stack.push()
     */
    public int pickUp(int[] order) {
        Stack<Integer> stack = new Stack<>();
        int index = 0;
 
        for (int i = 0; i < order.length; i++) {
            if (i+1 == order[index]) {
                index++;
                while (stack.size() != 0 && stack.peek() == order[index]) {
                    index++;
                    stack.pop();
                }
            } else {
                stack.push(i+1);
            }
        }
        return index;
    }
}

 

 

결과

 

 

주의사항

  ① 스택의 크기가 0일때의 분기처리