小鑫的算法之路：leetcode0027 移除元素

题目

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

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

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

示例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。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

解法1

在解题前，我们需要先明确如下几点：

1、什么是原地移除值为val的元素？将值为val的元素移动至数组末尾，或者修改为其他值都没有关系，因为不需考虑数组中超出新长度后面的元素；

2、题目明确说使用O(1)的空间复杂度解决问题；

3、元素的顺序可以修改。

该题目和leetcode第283题类似，设置第一个值为val所在的索引值为first_val_index，将其初始化为0，那么在区间[0, first_val_index)内的值均不为val，[first_val_index, size)内的值均为val。

代码如下：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int first_val_index = 0;
        int size = nums.size();
        // 区间[0, first_val_index)内的值均不为val, [first_val_index, size)内的值均为val
        for (int index = 0; index < size; ++index) {
            if (nums[index] != val) {
                nums[first_val_index] = nums[index];
                ++first_val_index;
            }
        }
        // 由于[first_val_index, size)内的值均为val，则长度为first_val_index
        return first_val_index;  
    }
};

时间复杂度为O(n)，空间复杂度为O(1)。执行结果为：

解法2

由于元素的顺序可以改变，那么考虑使用交换，如果左边的值为val，右边的值不为val，那么直接交换即可。可以使用双指针法，左指针初始索引值为0，右指针初始索引值为数组的长度减1。左边指针向右移动，如果值为val时停止移动，如果右指针此时值不为val，则交换，否则向左移动，直到找到一个不为val的值。注意如上操作均需要保证左指针的索引不大于右指针的索引。

代码如下：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int left = 0;
        int right = nums.size() - 1;
        // [0, left)区间内值不为val, (right, nums.size() - 1]区间内值为val
        while (left <= right) {
            // 如果左指针指向的值不为val，则左指针继续向右移动
            if (nums[left] != val) {
                ++left; 
                continue;
            }

            // 如果左指针指向的值为val, 此时右指针指向的值也同样为val，则右指针继续向左移动
            if (nums[right] == val) {
                --right;
                continue;
            }

            // 如果左指针指向的值为val, 此时右指针指向的值不为val，则交换左右指针指向的值
            // 并且左指针继续向右移动，右指针继续向左移动
            swap(nums[left], nums[right]);
            ++left;
            --right;
        }

        return left;  
    }
};

时间复杂度为O(n)，空间复杂度为O(1)。执行结果为：