Leetcode题解-数组相关-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。注意这五个元素可为任意顺序。你不需要考虑数组中超出新长度后面的元素。

leetcode链接

思路

此题有多种解决方案。
首先可以用暴力解决，即找到目标元素就把后面的所有元素往前移一位，重复此过程即可。此方法需要两个for循环嵌套，时间复杂度为O(n²)。

另一种方法是双指针法，双指针法又包含快慢指针法和相向指针法两种。

快慢指针法即通过快指针的移动完成元素的筛选（把目标元素排除），慢指针用来指向新数组元素（并没有创建新数组，只是为了和原数组做区别），最终的新数组的长度就由慢指针来决定。

相向指针法的核心思想是用尾部保留元素替换头部的目标元素。相向指针法和快慢指针不同的是，快慢指针是同一个方向移动，只是移动的速度不一样，而相向指针法顾名思义，是从不同的方向移动，一个指针从头移动到尾，一个指针从尾移动到头。头指针移动时，当发现目标（移除）元素，就用尾部的非目标元素替换，重复此过程即可。注意此方法返回的元素顺序和原来的顺序是不一样的。

本篇介绍两种双指针法的实现，两种方法的时间复杂度都是O(n)。

算法核心

1. 快慢指针法

1）初始条件，两个指针同时从数组头部开始移动，slowIndex=fastIndext=0。
2）当查找到目标元素，停止慢指针的移动（这就导致了一个指针移动的快，一个指针移动的慢）。
3）在循环过程中，快指针指向的元素（排除目标元素）都要覆盖慢指针指向的元素。
4）循环终止条件，快指针把所有元素都遍历完。
5）新数组大小由慢指针的值决定。

2. 相向指针法

1）初始条件，一个指针指向数组头部，另一个指针指向尾部第一个不是目标元素的位置。
2）移动头指针，当找到目标元素，用尾部保留元素覆盖（其实和元素互换类似）。注意尾部指针要更新，要移到下一个不是目标元素的位置。
3）循环终止条件，当头指针超过（大于）尾指针的时候。

代码 & 复杂度

1. 快慢指针法

public int removeElement(int[] nums, int val) {
		int slowIndex = 0;
		
		for(int fastIndex = 0; fastIndex < nums.length; fastIndex++) {
			if(nums[fastIndex] != val) {
				nums[slowIndex] = nums[fastIndex];
				slowIndex++;
			}
		}
		return slowIndex;

    }

时间复杂度: O(n)
空间复杂度: O(1)

2. 相向指针法

/**
	 * 27. 移除元素--相向双指针法（改变了元素的相对顺序）
	 */
public static int removeElement(int[] nums, int val) {
		int left = 0, right = nums.length - 1;
		
		while(right >= 0 && nums[right] == val) {
			// 注意数组越界的判断
			right--;
		}
		
		while(left <= right) {
			if(nums[left] == val) {
				nums[left] = nums[right];
				right--;
				while(right >= 0 && nums[right] == val) {
					right--;
				}
			}
			left++;
		}
		return left;
    }

时间复杂度: O(n)
空间复杂度: O(1)

注意点

1.while(right >= 0 && nums[right] == val) {
right–;
} 中一定注意right >= 0的判断，不然会有数组越界的异常。
2.当边界条件和指针移动比较懵时，画图更直观些。