【刷题之路】LeetCode 27. 移除元素

一、题目描述

原题连接： 27. 移除元素
题目描述：
给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。
不要使用额外的数组空间，你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。

示例 1：
输入： nums = [3,2,2,3], val = 3
输出： 2, nums = [2,2]
解释： 函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。你不需要考虑数组中超出新长度后面的元素。例如，函数返回的新长度为 2 ，而 nums = [2,2,3,3] 或 nums = [2,2,0,0]，也会被视作正确答案。

示例 2：
输入： nums = [0,1,2,2,3,0,4,2], val = 2
输出： 5, nums = [0,1,4,0,3]
解释： 函数应该返回新的长度 5, 并且 nums 中的前五个元素为 0, 1, 3, 0, 4。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

二、解题

1、方法1——暴力法

1.1、思路分析

我们首先能想到的就是暴力法，第一层循环遍历数组，找到val后停下，用第二层循环将后面的元素全都向前移动一位。

1.2、代码实现

有了以上分析，那我们写起代码来也就水到渠成了：

int removeElement1(int* nums, int numsSize, int val) {
	assert(nums);
	int i = 0;
	int j = 0;
	int n = 0; // 记录数组中val的个数
	for (i = 0; i < numsSize - n; i++) {
		if (nums[i] == val) {
			for (j = i; j < numsSize - n - 1; j++) {
				nums[j] = nums[j + 1];
			}
			i--; // 因为有可能在val后面又跟一个val，所以下一次在同一位置开始比较保险
			n++; // 找到了一个val
		}
	}
	return numsSize - n;
}

时间复杂度：O(n^2), n为数组元素个数
空间复杂度：O(1)，我们只需要用到常数级的额外空间

2、方法2——双指针

2.1、思路分析

很显然，去除val值后的数组的长度一定小于或等于原来数组的长度，所以我们可以直接把去掉val后的数组写到原来的数组中，这个过程可以通过两个指针来完成。
这里的“双指针”不同于以往的双指针，这里的“双指针”不再是左指针和右指针，而是赋值指针和遍历指针：
我们用一个give指针来控制赋值，每赋值成功give就往后走一步，用一个temp指针来遍历数组，如果temp指向的元素不等于val，则通过give指针将其存入数组中，若等于val，则temp继续向前走，give则不动。

2.2、代码实现

有了以上分析，那我们写起代码来也就水到渠成了：

int removeElement(int* nums, int numsSize, int val) {
	assert(nums);
	int* give = nums;
	int* temp = nums;
    int n = 0; // 记录数组中val的个数
	while (temp < nums + numsSize) {
		if (*temp == val) {
            n++;
			temp++;
		}
		else {
			*give = *temp;
			give++;
			temp++;
		}
	}
	return numsSize - n;
}

时间复杂度：O(n),其中n为序列的长度。我们只需要用temp遍历数组一次即可。
空间复杂度：O(1),我们只需要常数的空间保存若干变量。

3、方法3——双指针优化

3.1、思路分析

如果要移除的元素正好在数组的开头，例如{ 1, 2, 3, 4, 5 },当val为1时，我们需要把每一个元素都向前移动一位，好像比较麻烦。
但注意题目中所说的：“元素的顺序可以改变”。实际上我们可以直接将最后一个元素移动到开头，取代元素1，得到{ 5, 2, 3, 4 }，这样同样满足题目要求。这个优化在数组中val较少的时候非常有效。
我们让left指针指向数组的开头，right指针指向数组的末尾，left向右遍历，找到一个val后停下，此时将right指向的元素赋值到左指针指向的位置，然后右指针向左移动一位，如果赋值过来的元素恰好也等于val，则可以继续把right指向的元素赋值到left指向的位置，直到左指针指向的元素不等于val时，让left向右移动一位。
当做指针和右指针重合时，左右指针就遍历完了数组中所有的元素。

3.2、代码实现

有了以上思路，那我们写起代码来也就水到渠成了：

int removeElement3(int* nums, int numsSize, int val) {
	assert(nums);
	int left = 0, right = numsSize;
	while (left < right) {
		if (nums[left] == val) {
			nums[left] = nums[right - 1];
			right--;
		}
		else {
			left++;
		}
	}
	return left;
}

时间复杂度：O(n),n为数组元素个数，我们只需要遍历数组一次。
空间复杂度：O(1),我们只需要用到常数级的额外空间。