一、原地移除所有数值等于 val 的元素
LeetCode27.给你一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,并返回移除后数组的新长度。要求:不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并原地修改输入数组。元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
这是LeetCode上一道关于删除数组元素的题目,我们知道数组在删除的时候,从删除位置开始的所有元素都要向前移动,所以这题的关键是如果有很多值为val的元素的时候,如何避免反复向前移动呢? 使用我们的双指针就可以啦,而且有三种方式,我们一一道来。
方法一:快慢指针
核心思想是:定义两个指针slow和fast,slow指针之前的位置是数组的有效部分。
fast指针用来遍历整个数组,如果nums[fast]!=val,则将该位置的元素复制给slow指针所在的位置nums[slow]=nums[fast] 同时slow的位置向前一步,slow++。
反之,如果fast指针所在的位置是要删除的元素,则不进行其它操作slow留在当前位置,让fast指针继续向前遍历,直到nums[fast]!=val。
图示:
代码:
public static int removeElement(int[] nums, int val) {
int slow = 0;
//fast充当了快指针的角色
for (int fast = 0; fast < nums.length; fast++) {
if (nums[fast] != val) {
nums[slow] = nums[fast];
slow++;
}
}
//最后剩余元素的数量
return slow;
}
方法二:对撞指针
核心思想:定义两个指针,left=0在数组的开头,right=nums.length-1在数组末尾。
left指针的位置不是val就继续前进left++,遇到val就停下
相反,当right指针的位置是val就后退right--,遇到不是val的就停下。
当nums[left]==val和nums[fast]!=val时 ,我们就交换两个指针位置的值。
l两个指针相遇, left>right的时候,代码就执行结束了。
图示:
代码:
public int removeElement(int[] nums, int val) {
int right = nums.length - 1;
int left = 0;
for (left = 0; left <= right; ) {
if ((nums[left] == val) && (nums[right] != val)) {
//交换两个指针位置的值
int tmp = nums[left];
nums[left] = nums[right];
nums[right] = tmp;
}
if (nums[left] != val) left++;
if (nums[right] == val) right--;
}
return left ;
}
方法三:对撞+覆盖
在上面的基础上,当nums[left]等于val的时候,我们就将nums[right]位置的元素覆盖nums[left],继续循环,如果nums[left]等于val就继续覆盖,否则才让left++,这也是双指针的方法。
实现代码:
public int removeElement(int[] nums, int val) {
int right = nums.length-1;
for (int left = 0; left <= right; ) {
if (nums[left] == val) {
nums[left] = nums[right ];
right--;
} else {
left++;
}
}
return right+1;
}
二:删除有序数组中的重复项
LeetCode26 给你一个有序数组 nums ,请你原地删除重复出现的元素,使每个元素只出现一次 ,返回删除后数组的新长度。不要使用额外的数组空间,你必须在原地修改输入数组 并在使用 O(1) 额外空间的条件下完成。
同样,我们可以使用双指针的思想
定义两个指针slow和fast,从数组0索引开始。slow指针之前的是数组的有效位置。fast负责遍历整个数组。
当两个指针位置的值一样,就让fast继续向前遍历,slow留在原位置
两个指针的值不一样时,就让slow前面那个位置的值等于fast现在所指位置的值。
注意:由于slow是从0开始,而我们返回的slow是代表数组的有效位置,比如最后slow停在4索引这个位置,但是实际有效位置是5。
所以,我们最后要返回slow+1
图示:
代码:
public int removeDuplicates(int[] nums) {
int slow = 0;
for (int fast = 0; fast < nums.length; fast++) {
if (nums[slow] != nums[fast]) {
slow++;
nums[slow] = nums[fast];
}
}
return slow+1;
}