LeetCode之移除元素

题目简介

原题链接：https://leetcode-cn.com/problems/remove-element/

著作权归领扣网络所有。

题目描述：

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。

元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例：

给定 nums = [3,2,2,3], val = 3,

函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。

你不需要考虑数组中超出新长度后面的元素。

更多细节见原链接。

题解

#include <iostream>
#include <vector>

using namespace std;

int removeElement(vector<int>& nums, int val)
{
    vector<int>::iterator it;
    for(it=nums.begin();it!=nums.end();it++)
    {
        if(*it == val)           //判断是否为需要删除的元素
        {
            nums.erase(it);      //删除元素
            --it;                //注意在删除元素后，移动迭代器至上一个元素，没有此操作，会遗漏元素。
        }
    }
    return nums.size();
}

int main() {
    vector<int> arr = {0,1,2,2,3,0,4,2};
    int num = 2;
    int result = removeElement(arr, num);
    cout << result << endl;
    return 0;
}

稍微改进的题解

发现删除后自减，再自加有点傻，所以小小改进了一下。

int removeElement(vector<int>& nums, int val)
{
    vector<int>::iterator it;
    it = nums.begin();
    while(it!=nums.end())
    {
        if(*it == val)
        {
            nums.erase(it);
        }
        else
        {
            it++;
        }
    }
    return nums.size();
}