[Programmers Level2] 택배상자(C++)

※주의※
저의 풀이가 정답은 아닙니다.
다른 코드가 더 효율적이거나 좋을 수 있습니다.
언제나 다른 사람의 코드는 참고만 하시기 바랍니다.

[문제 풀이 사이트]

프로그래머스

[문제 설명]

영재는 택배상자를 트럭에 싣는 일을 합니다. 영재가 실어야 하는 택배상자는 크기가 모두 같으며 1번 상자부터 n번 상자까지 번호가 증가하는 순서대로 컨테이너 벨트에 일렬로 놓여 영재에게 전달됩니다. 컨테이너 벨트는 한 방향으로만 진행이 가능해서 벨트에 놓인 순서대로(1번 상자부터) 상자를 내릴 수 있습니다. 하지만 컨테이너 벨트에 놓인 순서대로 택배상자를 내려 바로 트럭에 싣게 되면 택배 기사님이 배달하는 순서와 택배상자가 실려 있는 순서가 맞지 않아 배달에 차질이 생깁니다. 따라서 택배 기사님이 미리 알려준 순서에 맞게 영재가 택배상자를 실어야 합니다.
만약 컨테이너 벨트의 맨 앞에 놓인 상자가 현재 트럭에 실어야 하는 순서가 아니라면 그 상자를 트럭에 실을 순서가 될 때까지 잠시 다른 곳에 보관해야 합니다. 하지만 고객의 물건을 함부로 땅에 둘 수 없어 보조 컨테이너 벨트를 추가로 설치하였습니다. 보조 컨테이너 벨트는 앞 뒤로 이동이 가능하지만 입구 외에 다른 면이 막혀 있어서 맨 앞의 상자만 뺄 수 있습니다(즉, 가장 마지막에 보조 컨테이너 벨트에 보관한 상자부터 꺼내게 됩니다). 보조 컨테이너 벨트를 이용해도 기사님이 원하는 순서대로 상자를 싣지 못 한다면, 더 이상 상자를 싣지 않습니다.
예를 들어, 영재가 5개의 상자를 실어야 하며, 택배 기사님이 알려준 순서가 기존의 컨테이너 벨트에 네 번째, 세 번째, 첫 번째, 두 번째, 다섯 번째 놓인 택배상자 순서인 경우, 영재는 우선 첫 번째, 두 번째, 세 번째 상자를 보조 컨테이너 벨트에 보관합니다. 그 후 네 번째 상자를 트럭에 싣고 보조 컨테이너 벨트에서 세 번째 상자 빼서 트럭에싣습니다. 다음으로 첫 번째 상자를 실어야 하지만 보조 컨테이너 벨트에서는 두 번째 상자를, 기존의 컨테이너 벨트에는 다섯 번째 상자를 꺼낼 수 있기 때문에 더이상의 상자는 실을 수 없습니다. 따라서 트럭에는 2개의 상자만 실리게 됩니다.
택배 기사님이 원하는 상자 순서를 나타내는 정수 배열 order가 주어졌을 때, 영재가 몇 개의 상자를 실을 수 있는지 return 하는 solution 함수를 완성하세요.

[제한 사항]

• 1 ≤ order의 길이 ≤ 1,000,000
• order는 1이상 order의 길이 이하의 모든 정수가 한번씩 등장합니다.
• order[i]는 기존의 컨테이너 벨트에 order[i]번째 상자를 i+1번째로 트럭에 실어야 함을 의미합니다.

[입출력 예]

order	result
[4, 3, 1, 2, 5]	2
[5, 4, 3, 2, 1]	5

[입출력 예 설명]

입출력 예 #1

• 문제 예시와 같습니다.

입출력 예 #2

• 모든 상자를 보조 컨테이너 벨트에 모두 넣고, 역순으로 하나씩 빼서 트럭에 싣습니다.

[문제 풀이]

나는 기존 컨테이너를 큐로, 보조 컨테이너를 스택으로 설정하였다.
큐는 FIFO으로 먼저 넣은 값이 먼저 나오는 자료 구조이고,
스택은 FILO으로 먼저 넣은 값이 나중에 나오는 자료 구조이다.
내가 작성한 알고리즘은 아래와 같다.

1. 초기 택배 상자를 기존 컨테이너에 셋팅한다.
➜ 큐에 택배 상자 번호를 넣어준다.

2. 보조 컨테이너(스택)의 top이 택배 상자를 싣는 순서와 같은지 확인한다.
➜ 보조 컨테이너(스택)의 top과 택배 상자를 싣는 순서가 같다면 실을 수 있는 상자의 개수를 1 증가 시켜준다. 3. 기존 컨테이너(큐)의 맨 앞 원소와 택배 상자를 싣는 순서를 비교한다.
3-1. 기존 컨테이너(큐)의 맨 앞 원소와 택배 상자를 싣는 순서와 같다면
➜ 실을 수 있는 상자의 개수를 1 증가 시켜준다.
3-2. 기존 컨테이너(큐)의 맨 앞 원소와 택배 상자를 싣는 순서와 다르다면
➜ 큐가 비어있지 않을 때와 큐의 맨 앞 원소가 택배 상자를 싣는 순서가 같지 않을 때까지
기존 컨테이너의 맨 앞 원소를 보조 컨테이너로 옮긴다.
3-2-1. 기존 컨테이너(큐)의 택배 상자를 보조 컨테이너(스택)로 옮기고 나서
➜ 기존 컨테이너의 맨 앞 원소가 싣는 순서와 같다면 실을 수 있는 상자의 개수를 1 증가 시켜준다.
3-2-2. 기존 컨테이너(큐)의 택배 상자를 보조 컨테이너(스택)로 옮기고 나서
➜ 기존 컨테이너(큐)의 맨 앞 원소가 싣는 순서와 다르다면 기존(큐), 보조 컨테이너(스택)에서 실을 수 있는 상자가 없기 때문에 그대로 상자의 개수를 리턴하며 함수를 종료시켜준다.

[최종 코드]

[GitHub]

GitHub - SmallPeanutPark/Programmers: Study Programmers for Coding Test

#include <string>
#include <vector>
#include <stack>
#include <iostream>
#include <queue>
using namespace std;

queue<int> q; // 기존 컨테이너
stack<int> st; // 보조 컨테이너

void setInitBox(int len) {
    // 초기 택배 상자를 기존 컨테이너에 셋팅
    for(int i = 0; i < len; ++i) {
        q.push(i+1); 
    }
}

int setWantBoxOrder(vector<int> order, int len) {
    int count = 0;
    bool ispass = false;
    for(int i = 0; i < len; ++i) {
        if(!st.empty()) {
            if(st.top() == order[i]) {
                st.pop();
                count += 1;
                ispass = false;
                continue;
            }
        }

        if(q.front() != order[i]) {
            // 택배 상자를 실어야할 순서가 아니라면 택배를 다른 곳으로 보관해야한다.
            while((!q.empty()) && (q.front() != order[i])) {
                st.push(q.front());
                q.pop();
            }
            if(q.front() == order[i]) {
                q.pop();
                count += 1;
                ispass = false;
            } else {
                // 기존 컨테이너에서 찾지못했을 때
                ispass = true;
            }
        } else {
            // 택배 상자를 실어야할 박스가 기존 컨테이너에 있을 때
            ispass = false;
            q.pop();
            count += 1;
        }

        if(ispass) break;
    }
    return count;
}

int solution(vector<int> order) {
    int gSize = order.size();
    setInitBox(gSize);
    int answer = setWantBoxOrder(order, gSize);
    return answer;
}