这一篇文章总结了Leetcode中26题和27题的解法,主要是对双指针法的归纳。下面先来看看题目。
删除排序数组中的重复项
题目大意
给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
示例
给定数组 nums = [1,1,2],
函数应该返回新的长度 2, 并且原数组 nums 的前两个元素被修改为 1, 2。
你不需要考虑数组中超出新长度后面的元素。
给定 nums = [0,0,1,1,1,2,2,3,3,4],
函数应该返回新的长度 5, 并且原数组 nums 的前五个元素被修改为 0, 1, 2, 3, 4。
你不需要考虑数组中超出新长度后面的元素。
注意: 题目所说的“不需要考虑数组中超出新长度后面”的元素,意思就是可以通过覆盖的方法实现。
思路——双指针法
设置一个快指针j和一个慢指针i,快指针移动的范围为[0,len-1],在每一轮循环结束后++j。当快指针指向了数组末尾则算法结束。
代码
class Solution {
public int removeDuplicates(int[] nums) {
int len = nums.length;
if(len==0) return 0;
int i = 0;
for(int j=1;j<len;j++) {
if(nums[i]!=nums[j]) {
nums[i+1]=nums[j];
++i;
}
}
return i+1;
}
}
注意:双指针法要特别注意数组为空的时候,是否能统一返回,若不能,在算法开始时首先返回长度为0的情况。
移除元素
接下来看与上题非常相似的27题,移除指定的元素。
题目大意
给定一个数组 nums 和一个值 val,你需要原地移除所有数值等于 val 的元素,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在原地修改输入数组并在使用 O(1) 额外空间的条件下完成。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素。
示例
给定 nums = [3,2,2,3], val = 3,
函数应该返回新的长度 2, 并且 nums 中的前两个元素均为 2。
你不需要考虑数组中超出新长度后面的元素。
代码
class Solution {
public int removeElement(int[] nums, int val) {
int len = nums.length;
int i = 0;
for(int j = 0;j<len;j++) {
if(nums[j]!=val) {
nums[i] = nums[j];
++i;
}
}
return i;
}
}
总结:比较两个双指针法的模板,可以看到循环体中,均是为慢指针递增的情况写if语句,快指针不论什么情况都会++j。