【LeetCode】27.移除元素

27.移除元素

解题思路

本题最容易想到的方法就是通过遍历数组来找到val，然后删除(这里的删除指的是用后面的元素来覆盖)元素，如下方法一。

方法一：暴力求解法

 class Solution {
 public:
     int removeElement(vector<int>& nums, int val) {
         int size = nums.size();
         for (int i = 0; i < size; ++i)
         {
             if (nums[i] == val)
             {
                 for (int j = i; j < size - 1; ++j)
                 {
                     nums[j] = nums[j + 1];
                 }
                 --size;//
                 --i;//下面有解释为什么需要这一步
             }
         }
         return size;
     }
 };

代码很容易写，值得稍加思考的是为什么需要再--i

首先上报错！

我们可以清楚的看到输入有两个连续的2，这就是没有--i导致的，由于后面的数据都向前移了一位，所以这时候我们需要再重新判断一下nums[i]，所以--i这一步必不可少。

方法二：快慢指针法

fast是快指针，用来寻找新数组中的元素；slow是慢指针，用来更新旧数组。

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];//这一步可能不好理解，下面解释
            }

        }
        return slow;
    }
};

换个角度其实快慢指针法非常容易理解，想象一下，如果我们新开一个数组，用slow当做新数组的下边，将fast中的目标值赋给新数组，是不是就容易理解多了，如下图解：

当然，题目要求空间复杂度为O（1），所以我们不这样做，而是直接把原数组当做新数组，只在原数组上做改动就可以了。