LeetCode_27 移除元素

1、题目：移除元素

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

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

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

2、解题思路

利用双指针法，快指针指向当前将要处理的元素，慢指针指向下一个将要赋值的位置：如果快指针指向的元素不等于val，它一定是输出数组的一个元素，我们就将快指针指向的元素复制到慢指针位置，然后将快慢指针同时右移；如果快指针指向的元素等于val，它不能在输出数组里，此时慢指针不动，快指针右移一位。

由于此题的要求中指出可以改变原数组中元素的相对位置，所以有优化的双指针的方法：如果左指针left 指向的元素等于val，此时将右指针right 指向的元素复制到左指针left 的位置，然后右指针right 左移一位。如果赋值过来的元素恰好也等于val，可以继续把右指针right 指向的元素的值赋值过来（左指针left 指向的等于val 的元素的位置继续被覆盖），直到左指针指向的元素的值不等于val 为止。当左指针left 和右指针right 重合的时候，左右指针遍历完数组中所有的元素。这样的方法两个指针在最坏的情况下合起来只遍历了数组一次。与方法一不同的是，方法二避免了需要保留的元素的重复赋值操作。

3、代码

//方法一 从左往右进行双指针
class Solution 
{
public:
    int removeElement(vector<int>& nums, int val) 
    {
        int n = nums.size();
        int left = 0;
        for (int right = 0; right < n; right++) 
        {
            if (nums[right] != val) 
            {
                nums[left] = nums[right];
                left++;
            }
        }
        return left;
    }
};

//方法二 双指针优化
class Solution 
{
public:
    int removeElement(vector<int>& nums, int val) 
    {
        int left = 0, right = nums.size();
        while (left < right) 
        {
            if (nums[left] == val) 
            {
                nums[left] = nums[right - 1];
                right--;
            } 
            else 
            {
                left++;
            }
        }
        return left;
    }
};