leetcode-数组-简单-移除元素

题目

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/remove-element
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

给定 nums = [0,1,2,2,3,0,4,2], val = 2,
函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。
注意这五个元素可为任意顺序。
你不需要考虑数组中超出新长度后面的元素。

解答

这个题目和删除重复项基本是一样的，没有太多要分析的。

方法一：

找到元素，把后边的依次往前移动即可

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int count = 0;
        for (int i = 0; i < nums.size() - count;) {
            if (val == nums[i]) {
                for (int j = i; j < nums.size() - 1 - count; j++) {
                    nums[j] = nums[j + 1];
                }
                count++;
            } else {
                i++;
            }
        }
        return nums.size() - count;
    }
};

方法二：

这个方法还是处理不相同的元素

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

可以对比一下两个方法的差异，

方法一虽然进行了两重循环，但是只有在元素相等的时候才进行元素移动，其他的数据并不会进入if判断内部。

而方法二虽然只有一次循环 ，但是大部分情况下if判断是满足的，元素的拷贝次数还是比较多的。