27. 移除元素

27. 移除元素

题目描述

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

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

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

示例1：

给定 nums = [3,2,2,3], val = 3,

函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。

你不需要考虑数组中超出新长度后面的元素。

示例2：

给定 nums = [0,1,2,2,3,0,4,2], val = 2,

函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。

注意这五个元素可为任意顺序。

你不需要考虑数组中超出新长度后面的元素。

---

题解：

双指针，不过写法有两种：

法一：

使用快慢指针，i 是慢指针，j 是快指针。j 从 0 开始遍历 nums ：

1. 若 nums[j] == val ，跳过该元素，即 ++j
2. 否则 nums[i++] = nums[j++]

法一代码：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int n = nums.size();
        if ( !n ) return 0;
        if ( n == 1 ) return !( nums[0] == val );
        int i = 0;
        for ( int j = 0; j < n; ++j ) {
            if ( nums[j] == val ) continue;
            nums[i++] = nums[j];
        }
        return i;
    }
};
/*
时间：0ms，击败：100.00%
内存：8.4MB，击败：90.21%
*/

法二：

在某些情况下，法一会出现一些不必要的操作：

• 若 nums=[1, 2, 3, 4, 5], val = 5 ，则会对前面 4 个元素做不必要的复制操作
• 若 nums=[1, 2, 3, 4, 5], val = 1 ，没有必要将后面4个元素往左移动一位

可以考虑将为 val 的元素移动到尾部，并删除尾部。

仍是使用两个指针 i 和 j ，分别指向第一个和最后一个元素，当 i <= j 时：

• 若 nums[i] == val ，则 swap( nums[i], nums[j--] ) ，这种情况下，有可能 nums[j] == val ，所以 i 不动
• 若 nums[i] != val ，则 ++i

最后返回 i 即可。

法二代码：

class Solution {
public:
    int removeElement(vector<int>& nums, int val) {
        int n = nums.size();
        if ( !n ) return 0;
        if ( n == 1 ) return nums[0] != val ;
        int i = 0, j = n - 1;
        while ( i <= j ) {
            if ( nums[i] == val ) swap( nums[i], nums[j--] );
            else ++i;
        }
        return i;
    }
};
/*
时间：0ms，击败：100.00%
内存：8.6MB，击败：88.44%
*/

好像内存消耗更大了2333