[Programmers Lv2, Java] 택배 상자

영재는 택배상자를 트럭에 싣는 일을 합니다. 영재가 실어야 하는 택배상자는 크기가 모두 같으며 1번 상자부터 n번 상자까지 번호가 증가하는 순서대로 컨테이너 벨트에 일렬로 놓여 영재에게 전달됩니다. 컨테이너 벨트는 한 방향으로만 진행이 가능해서 벨트에 놓인 순서대로(1번 상자부터) 상자를 내릴 수 있습니다. 하지만 컨테이너 벨트에 놓인 순서대로 택배상자를 내려 바로 트럭에 싣게 되면 택배 기사님이 배달하는 순서와 택배상자가 실려 있는 순서가 맞지 않아 배달에 차질이 생깁니다. 따라서 택배 기사님이 미리 알려준 순서에 맞게 영재가 택배상자를 실어야 합니다.

만약 컨테이너 벨트의 맨 앞에 놓인 상자가 현재 트럭에 실어야 하는 순서가 아니라면 그 상자를 트럭에 실을 순서가 될 때까지 잠시 다른 곳에 보관해야 합니다. 하지만 고객의 물건을 함부로 땅에 둘 수 없어 보조 컨테이너 벨트를 추가로 설치하였습니다. 보조 컨테이너 벨트는 앞 뒤로 이동이 가능하지만 입구 외에 다른 면이 막혀 있어서 맨 앞의 상자만 뺄 수 있습니다(즉, 가장 마지막에 보조 컨테이너 벨트에 보관한 상자부터 꺼내게 됩니다). 보조 컨테이너 벨트를 이용해도 기사님이 원하는 순서대로 상자를 싣지 못 한다면, 더 이상 상자를 싣지 않습니다.

예를 들어, 영재가 5개의 상자를 실어야 하며, 택배 기사님이 알려준 순서가 기존의 컨테이너 벨트에 네 번째, 세 번째, 첫 번째, 두 번째, 다섯 번째 놓인 택배상자 순서인 경우, 영재는 우선 첫 번째, 두 번째, 세 번째 상자를 보조 컨테이너 벨트에 보관합니다. 그 후 네 번째 상자를 트럭에 싣고 보조 컨테이너 벨트에서 세 번째 상자 빼서 트럭에싣습니다. 다음으로 첫 번째 상자를 실어야 하지만 보조 컨테이너 벨트에서는 두 번째 상자를, 기존의 컨테이너 벨트에는 다섯 번째 상자를 꺼낼 수 있기 때문에 더이상의 상자는 실을 수 없습니다. 따라서 트럭에는 2개의 상자만 실리게 됩니다.

택배 기사님이 원하는 상자 순서를 나타내는 정수 배열 order가 주어졌을 때, 영재가 몇 개의 상자를 실을 수 있는지 return 하는 solution 함수를 완성하세요.

---

이건 문제만 봐도 쉬웠다. 근데 보통 컨베이어벨트라고 하지않나? Conveyor belt인가 그럴텐데...

아무튼

메인 컨베이어 벨트는 1번상자부터 차례대로 꺼낼 수 있는 Queue 구조,

서브 컨베이어 벨트는 가장 최근에 올린걸 가장 먼저 꺼낼 수 있는 Stack 구조이다.

Queue같은 경우에는 직접 구현할 필요 없이 1번 상자부터 order.length번 상자 까지 검사하면 Queue와 같고,

Stack은 java.util.Stack을 이용했다.

 

    public int solution(int[] order) {
        int answer = 0;

        Stack<Integer> subConveyor = new Stack<>();
        for(int i = 1 ; i <= order.length ; i++) {
            int canPlace = order[answer];

            // 메인 컨베이어의 제일 앞에 있는 상자가 올릴 수 있는 상자라면 올린다.
            if(canPlace == i) {
                answer++;
                continue;
            } else if(!subConveyor.empty()) {
                int peek = subConveyor.peek();
                // 서브 컨베이어의 제일 앞에 있는 상자가 올릴 수 있는 상자라면 올린다.
                if (canPlace == peek) {
                    answer++;
                    i--;
                    subConveyor.pop();
                    continue;
                }
            }
            // 올릴 수 있는 상자가 없다면 서브 컨베이어로 옮긴다.
            subConveyor.push(i);
        }

        // 메인 컨베이어의 상자는 모두 끝나고 서브 컨베이어만 남은 경우 탐색
        while(!subConveyor.empty() && order[answer] == subConveyor.pop()) answer++;

        return answer;
    }

첫 for문을 통해 메인컨베이어를 모두 탐색하고,

뒤쪽 while을 통해 서브컨베이어에 남은것이 있는지 까지 마저 탐색한다.

 

나온지 얼마 안 된문제(완료한사람 447명)라서 그런지 정답률(34%)에 비해 굉장히 쉬운문제였다.