leetcode—数组—移除元素

移除元素

题目类型： 数组
题目来源： leetcode传送门

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

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

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

说明：
为什么返回数值是整数，但输出的答案是数组呢?

请注意，输入数组是以「引用」方式传递的，这意味着在函数里修改输入数组对于调用者是可见的。

你可以想象内部操作如下:

// nums 是以“引用”方式传递的。也就是说，不对实参作任何拷贝
int len = removeElement(nums, val);

// 在函数里修改输入数组对于调用者是可见的。
// 根据你的函数返回的长度, 它会打印出数组中 该长度范围内 的所有元素。
for (int i = 0; i < len; i++) {
print(nums[i]);
}

示例1：

输入：nums = [3,2,2,3], val = 3
输出：2, nums = [2,2]
解释：函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为 2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案。

示例2：

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

提示：

• 0 <= nums.length <= 100
• 0 <= nums[i] <= 50
• 0 <= val <= 100

方法一（自己最开始的做法，思路比较复杂）：

看到这题自己的想法就是利用一个类似next数组的东西来解决，这个next数组是嵌入在原数组之中的。一共有3个指针，slow，lastSlow，fast。fast指针就是遍历数组的指针，slow指针代表从左向右第一个可以向其中填入新元素的位置（即这个位置当前是空的），slow位置中的元素是下一个空位置的坐标，下一个空位置的元素是下下个空位置的坐标，依次类推，最后一个空位置的元素为-1。每次fast的位置如果不是val且slow不等于-1，则将fast中的元素添加到slow除，更新slow为slow中记录的下一个空位置的坐标，并将lastSlow中的值改为fast，并将lastSlow赋为fast，fast中的元素设为-1，total++；fast位置如果不是val且slow是-1，则total++；如果fast中的元素是val且slow不是-1，则将lastSlow中的值改为fast，fast中的值改为-1；如果fast中的元素是val且slow是-1，则将fast中的值改为-1，slow和lastSlow都改为fast。听起来可能有些复杂，用一个例子来说明：
数组为nums=[0,1,2,2,3,0,4,2]，val=2

fast=0， slow=-1，lastSlow=-1，total=0

坐标	0	1	2	3	4	5	6	7
元素	0	1	2	2	3	0	4	2

fast=1，slow=-1，lastSlow=-1，total=1

坐标	0	1	2	3	4	5	6	7
元素	0	1	2	2	3	0	4	2

fast=2，slow=2，lastSlow=2，total=2

坐标	0	1	2	3	4	5	6	7
元素	0	1	-1	2	3	0	4	2

fast=3，slow=2，lastSlow=3，total=2

坐标	0	1	2	3	4	5	6	7
元素	0	1	3	-1	3	0	4	2

fast=4，slow=3，lastSlow=4，total=2

坐标	0	1	2	3	4	5	6	7
元素	0	1	3	4	-1	0	4	2

fast=5，slow=4，lastSlow=5，total=3

坐标	0	1	2	3	4	5	6	7
元素	0	1	3	0	5	-1	4	2

fast=6，slow=5，lastSlow=6，total=4

坐标	0	1	2	3	4	5	6	7
元素	0	1	3	0	4	6	-1	2

fast=7，slow=5，lastSlow=7，total=5

坐标	0	1	2	3	4	5	6	7
元素	0	1	3	0	4	6	7	-1

此时就完成了修改，返回的total是5。

代码如下：

int removeElement(vector<int>& nums, int val) {
        int slow = -1;
        int total = 0;
        int temp = -1;
        int lastSlow = -1;
        for(int fast = 0; fast < nums.size(); fast ++)
        {
            if(nums[fast] == val && slow == -1)
            {
                slow = fast;
                lastSlow = fast;
                nums[fast] = -1;
            }
            else if(nums[fast] == val && slow != -1)
            {
                nums[fast] = -1;
                nums[lastSlow] = fast;
                lastSlow = fast;
            }
            else if(nums[fast] != val && slow == -1)
            {
                total ++;
            }
            else if(nums[fast] != val && slow != -1)
            {
                if(nums[slow] != -1)
                {
                    temp = slow;
                    slow = nums[slow];
                    nums[temp] = nums[fast];
                    nums[fast] = -1;
                    nums[lastSlow] = fast;
                    lastSlow = fast;
                    total ++;
                }
                else
                {
                    nums[slow] = nums[fast];
                    nums[fast] = -1;
                    slow = fast;
                    lastSlow = fast;
                    total ++;
                }
            }
        }

        return total;
    }

这题的难度标签是简单，但是上面这种做法我认为不可能是一道简单题的做法，所以我去看了一下题解，发现我想的实在是太太太太太太太复杂了，下面就说一下简单的做法。

方法二：

还记得之前我说过一个叫做双指针的方法，这题利用双指针就会很方便（而不是用快慢指针）

一个左指针left，一个右指针right，左指针永远指向元素是val的位置，右指针如果遇到不是val的就将其元素放到左指针的位置，左指针再继续向右找val，这样就可以得到答案了。答案是最后左指针的位置加一。

代码如下：
由于考虑到nums的size可能为0，所以需要特判

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

        if(nums[left] != val)
            total ++;

        return total;
    }