力扣27. 移除元素-不一样的想法

题目描述:

给你一个数组 nums 和一个值 val,你需要“原地”移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

题目连接-力扣27. 移除元素

题目其实想的就是给出移除后的个数,然后使用个数遍历给的数组
所以一般的思路就是使用双指针将等于val的值依次移到数组末尾

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int left=0,right=nums.size()-1;
        int res=0;
        while(1){
            if(left>right)return left;
            if(nums[left]!=val){
                left++;
                res++;
            }else if(nums[right]==val){
                right--;
            }else{
                nums[left]=nums[right];
                nums[right]=val;
                left++;
                right--;
            }
        }
        return 0;
    }
};

但我想的思路就是删除就删除呗,使用vector容器的删除函数直接删除就是了,其实也是很好理解(当然也顺便学习了容器的用法)
注意:it=nums.erase(it)就是vector容器的删除指定元素的函数,返回的是删除之后它将返回下一个位置的迭代器
所以当删除时其实就相当于更新了,所以要分情况

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int n=0;
        for(vector<int>::iterator it=nums.begin();it!=nums.end();){
            if(*it==val)it=nums.erase(it);
            else{
                it++;
                n++;
            }
            if(it==nums.end())break;
        }
        return n;
    }
};

上一篇:登录时生成验证码图片校验


下一篇:函数计算持续交付入门:云效+FC实现 简单IP查询工具