Level 0. 빈 배열에 추가, 삭제하기

Level 0. 빈 배열에 추가, 삭제하기

 

 

for문을 사용하지 않고 vector::insert(), vector::erase() 함수를 사용해도 된다.

 

#include <iostream>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr, vector<bool> flag)
{
    size_t size = arr.size();
    vector<int> answer;

    for (int i = 0; i < size; ++i)
    {
        if (flag[i])
        {
            vector<int> v(arr[i] * 2, arr[i]);

            answer.insert(answer.end(), v.begin(), v.end());
        }
        else
        {
            answer.erase(answer.end() - arr[i], answer.end());
        }
    }

    return answer;
}