给定一个数组arr和一个目标数targrt,删除所有目标值元素,并返回新长度

/***********************************************************************************
**description:给定一个数组arr和一个目标数targrt,删除所有目标值元素,并返回新长度
**            注:返回的新数组顺序不做要求
***********************************************************************************/
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
//方法一:利用一个index标记新数组长度,遍历原数组,当元素不等于target时存入新数组
//时间复杂度O(n),空间复杂度O(1)
int removeElement_1(vector<int> &arr, int target)
{
    int index = 0;
    for (int i = 0; i < arr.size(); i++)
    {
        if (arr[i] != target)
        {
            arr[index++] = arr[i];
        }
    }
    return index;
}

//方法二:直接利用库函数,remove()会返回一个指向新数组结尾的迭代器,remove在头文件algorithm中
//时间复杂度O(n),空间复杂度O(1)
int removeElement_2(vector<int> &arr, int target)
{
    return distance(arr.begin(),remove(arr.begin(),arr.end(),target));
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章