[Leetcode] Remove Element

描述

Given an array and a value, remove all instances of that value in place and return the new length.

Do not allocate extra space for another array, you must do this in place with constant memory.

The order of elements can be changed. It doesn’t matter what you leave beyond the new length.

Example:
Given input array nums = [3,2,2,3], val = 3

Your function should return length = 2, with the first two elements of nums being 2.

分析

去除数组中值为 val 的元素，并返回新数组的长度。

使用双指针的方法[代码1]，开始两个指针均指向原数组起始位置，指针1用于遍历原数组，当指针1所指的元素值不为 val 时，赋值给指针2所指的元素，指针2后移一个单位，重复以上过程直到指针1遍历完毕，时间复杂度 $O(n)$。

另外一种思路是[代码2]，因为可以不考虑新数组后面的值，可以利用一个指针遍历数组，当指针所指的元素值为 val 时，将数组最后一个非 val 元素赋值给它，然后减小数组的长度，时间复杂度 $O(n)$。

代码1

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

代码2

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