給定一個數組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));
}

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