1 题目:移除元素
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O ( 1 ) O(1) O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
题目来源:LeetCode27.移除元素
1.1 方法一:快慢双指针
思想:定义两个指针 slow 和 fast,初始时让 slow 和 fast 指向数组的第 1 个元素,在遍历数组时,fast 遇到数值等于 val 的元素时,让 slow 停下,而其继续前进直到找到数值不等于 val 的元素,再用 nums[fast] 的值来覆盖 nums[slow] 的值。
图示如下:
使用快慢双指针移除元素后,留下的元素顺序跟原始顺序保持一致。
Java 代码实现如下:
public int removeElement(int[] nums, int val) {
int slow = 0;
for(int fast=0; fast<nums.length; fast++){
if(nums[fast] != val){
nums[slow] = nums[fast];
slow++;
}
}
return slow;
}
1.2 方法二:对撞双指针
思想:定义两个指针 left 和 right ,初始时 left 指向数组的第 1 个元素, right 指向数组的最后一个元素,在遍历数组时,通过 right 从右边找到元素值不等于 val 的元素来替换左边 left 指向的元素值为 val 的元素。
图示如下:
使用对撞双指针移除元素后,留下的元素顺序跟原始顺序可能不一致。
Java 代码实现如下:
public int removeElement(int[] nums, int val) {
int left = 0;
int right = nums.length - 1;
while(left <= right){
if(nums[left]==val && nums[right]!=val){
int temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
}
if(nums[left] != val){
left++;
}
if(nums[right] == val){
right--;
}
}
return left;
}
1.3 方法三:对撞双指针 + 覆盖
思想:定义两个指针 left 和 right ,初始时 left 指向数组的第 1 个元素, right 指向数组的最后一个元素,在遍历数组的过程中,当 nums[left] == val 时,就用 nums[right] 的值来覆盖 nums[left] 的值。
图示如下:
Java 代码实现如下:
public int removeElement(int[] nums, int val) {
int left = 0;
int right = nums.length - 1;
while(left <= right){
if(nums[left] == val){
nums[left] = nums[right];
right--;
}else{
left++;
}
}
return left;
}
2 题目:删除有序数组中的重复项
给你一个 升序排列 的数组 nums
,请你 原地 删除重复出现的元素,使每个元素 只出现一次 ,返回删除后数组的新长度。元素的 相对顺序 应该保持 一致 。然后返回 nums
中唯一元素的个数。
思想:定义两个指针 slow 和 fast,初始时让 slow 和 fast 指向数组的第 2 个元素,遍历数组时比较 nums[fast] 和 nums[slow-1] ,当 nums[fast] == nums[slow-1] 时,让 slow 停下,而让 fast 继续前进直到 nums[fast] != nums[slow-1] 时,用 nums[fast] 的值覆盖 nums[slow] 的值。
图示如下:
问:为什么初始时让 slow 和 fast 指向数组的第 2 个元素?
答:我们假设第 1 个元素总是非重复的,因为它没有其他元素可以与之比较。
Java 代码实现如下:
public int removeDuplicates(int[] nums) {
int slow = 1;
for(int fast=1; fast<nums.length; fast++){
if(nums[fast] != nums[slow-1]){
nums[slow] = nums[fast];
slow++;
}
}
return slow;
}
3 总结
本文主要运用双指针解决了 LeetCode27.移除元素和 LeetCode26.删除有序数组中的重复项这两道问题。双指针法是一种高效、简单且不需要额外空间的算法技巧。双指针法通常只需要在原始数组上进行操作,而不需要使用额外的数据结构,因此节省了空间复杂度,而且双指针法通常在一次遍历中完成问题的解决,无需多次遍历数组,效率较高。