双指针
双指针是一种操作数组还有链表非常好用的技巧,本文会涉及到的问题有数组移除指定元素、排序数组移除重复元素。
1.数组移除指定元素(来源leetcode)
给你一个数组 nums 和一个值 val,你需要 原地 移除所有数值等于 val 的元素,并返回移除后数组的新长度。
不要使用额外的数组空间,你必须仅使用 O(1) 额外空间并 原地 修改输入数组。
元素的顺序可以改变。你不需要考虑数组中超出新长度后面的元素
思路:
如果没有限制,那么这个题目会非常简单,但是总是要给自己来点挑战的嘛,这样不知道什么时候就会用上了。
这个题目只使用一个指针肯定不可以完成任务,因为找到了不是val的元素之后,却不知道要往哪里存放,所以需要在拿一个指针来存放地址,这个方法
般被叫做快慢指针
使用两个指针slow,fast,都从数组头开始遍历,如果fast指向的元素不是需要删除的元素,那么就把他放到slow的位置上,同时,fast和slow向右移动一格,否则只有fast移动,接下来证明一下这样为什么可以完成任务?
如果fast和slow指向同一个元素,这个元素只有两种可能性,要么是val要么不是val,如果不是val,那么fast和slow都向后移动,如果是val,那么只有fast向后移动,而fast指到非val的值的时候,就会覆盖slow的值,这个覆盖会不会导致元素丢失呢?
不会,因为fast和slow指向同一位置的时候,覆盖不会影响数组的值,fast的slow指向不同位置的时候,slow的值一定是val的值,覆盖val就是题目的要求,所以不会。
代码:
class Solution {
public int removeElement(int[] nums, int val) {
int fast = 0;
int slow = 0;
while(fast < nums.length){
if(nums[fast] == val){
fast++;
}else{
int t = nums[slow];
nums[slow] = nums[fast];
nums[fast] = t;
fast++;
slow++;
}
}
return slow;
}
}
2.排序数组移除重复元素(来源leetcode)
给定一个排序数组,你需要在 原地 删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度。
不要使用额外的数组空间,你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成。
同样使用快慢指针完成
代码:
class Solution {
public int removeDuplicates(int[] nums) {
if(nums.length == 0) return 0;
if(nums.length == 1) return 1;
int slow = 0;
int fast = 1;
while(fast < nums.length){
if(nums[fast] == nums[slow]){
fast++;
}else{
nums[slow + 1] = nums[fast];
slow++;
fast++;
}
}
return slow + 1;
}
}