[알고리즘 + 자료구조]/[프로그래머스]

프로그래머스 택배 상자 C++

Hevton 2023. 8. 10. 11:54
반응형

 

 

규칙을 찾아서 하면 될 것 같았는데

직접 구현하는 수 밖에 없겠다.

#include <string>
#include <vector>
#include <stack>

using namespace std;

int solution(vector<int> order) {
    int answer = 0;
    int size = order.size();
    
    stack<int> stk;
    
    for(int i = 1; i <= size; i++) {
        stk.push(i);
        
        while(!stk.empty() && stk.top() == order[answer]) {
            stk.pop();
            answer++;
        }
        
    }
    
    return answer;
}

스택 하나만 있어도 가능하다

반응형