LeetCode之移除元素

Posted O(∩_∩)O

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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();
}

以上是关于LeetCode之移除元素的主要内容,如果未能解决你的问题,请参考以下文章

leetcode27之移除元素

简化条件表达式之移除控制标记(Remove Control Flag)

egret之移除带参数的监听事件

LeetCode27. 移除元素

LeetCode.27 移除元素

pandas相同元素不同顺序合并之移形换位