프로그래머스 - 배열의 원소 삭제하기 - C++

https://school.programmers.co.kr/learn/courses/30/lessons/181844

프로그래머스

 

 

#include <string>
#include <vector>
#include<algorithm>
using namespace std;

vector<int> solution(vector<int> arr, vector<int> delete_list) {
    vector<int> answer;
     for(int i=0;i<delete_list.size();i++)
         arr.erase(remove(arr.begin(), arr.end(), delete_list[i]), arr.end());
    return arr;
}

설명

string.erase , remove()

삭제할 원소들이 들어있는 delete_list를 for문으로 돌며 1개씩 arr에서 전부 삭제한다. 

 

이 때, erase함수가 사용된 코드줄을 보면 remove를 함수는 vector의 크기를 변경할 수 없다. 단지 삭제할 값 뒤에 있는 원소들을 한칸 씩 땡겨온다. 그리고 마지막으로 삭제되어야할 그 원소의 위치값을 반환한다. 

그러면 erase함수를 통해 그 위치부터 end까지 모든 원소를 삭제한다.