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

문제설명

정수 배열 arr과 delete_list가 있습니다.
arr의 원소 중 delete_list의 원소를 모두 삭제하고
남은 원소들은 기존의 arr에 있던 순서를 유지한 배열을
return 하는 solution 함수를 작성해 주세요.

제한사항

• 1 ≤ arr의 길이 ≤ 100
• 1 ≤ arr의 원소 ≤ 1,000
• arr의 원소는 모두 서로 다릅니다.
• 1 ≤ delete_list의 길이 ≤ 100
• 1 ≤ delete_list의 원소 ≤ 1,000
• delete_list의 원소는 모두 서로 다릅니다.

입출력 예

arr	delete_list	result
[293, 1000, 395, 678, 94]	[94, 777, 104, 1000, 1, 12]	[293, 395, 678]
[110, 66, 439, 785, 1]	[377, 823, 119, 43]	[110, 66, 439, 785, 1]

풀이

정수 배열 arr에서 delete_list에 있는 원소를 삭제하는 문제입니다. find 함수를 이용하여 원소를 찾고 삭제하는 로직을 구현하여 문제를 풀 수 있었습니다.

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

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

결과