力扣--27 移除元素 C++

题目描述
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
思路
1.双指针法
一个指针用于寻找,跳过值为val的位置。一个指针用于放入非val的值。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int slow=0;
        for(int fast=0;fast<nums.size();fast++)
        {
            if(nums[fast]!=val)
            {
                nums[slow]=nums[fast];
                slow++;
            }
        }
        return slow;
    }
};

2.remove和erase搭配用法
remove负责将所有非val的元素放到vector的首部,值为val的元素放到vector的尾部,返回新的end()值(相当于val部分的begin)erase负责将尾部去除。

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        nums.erase(remove(nums.begin(),nums.end(),val),nums.end());
        return nums.size();
    }
};

关于remove和erase的区别可以参考博文:https://blog.csdn.net/xzymmd/article/details/83652726

发布了50 篇原创文章 · 获赞 30 · 访问量 9177

猜你喜欢

转载自blog.csdn.net/qq_42913794/article/details/104026252