LeetCode 27. Remove Element

题目

分析 && 代码

和前一道LeetCode 26. Remove Duplicates from Sorted Array一样的思路，不过它的follow-up还是有点意思。按照26题的思路，可以写出下面的代码：

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

follow-up是考虑当待删除的元素很少的时候：
不难发现，这个算法对nums所有的元素都检查了一遍，如果对于一个10w大小的nums，要删除1，而且只在最后一个位置是1其他都不是，那这个代码要要复制10w - 1次，i和len加起来变化了20w-1次。
就这一点点小地方，可以做一定优化。
我们可以让指针双向的走，单向，单向要走20w步，而双向可以只走一半，可以这么做的原因是题目说明元素顺序可以变化，来看代码就清楚了：

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

这样遍历结束i和n加起来走了n步，就这么一点点优化，两份代码交上去时间没有区别…当作开拓思维吧