对撞指针技巧
对撞指针技巧
我们以LeetCode的一道题目来讲解一下对撞指针;
LeetCode第27题移除元素,链接如下:
https://leetcode.cn/problems/remove-element
如果使用快慢指针
如果使用快慢指针,将会有大量的后面元素赋值给前面元素的操作,有没有其他不进行这么多次交换的思路呢?
对撞指针
我们的目的就是:将需要删除掉的元素放到数组的最后。
实际上,当我们遇到要删除的元素时,直接让数组后面的元素赋给这个元素即可。
这样的话,操作数量就会变少了。
那我们就需要定义一个指针,从后往前移动。
对撞指针思想
定义left和right两个指针,分别指向数组最开始和最后位置。
[0...left - 1]区间:不需要删除的区域
[left...right]区间:未处理的区域
[right + 1... n - 1]区间:需要删除的区域
- 如果left指向的元素值不等于待删除的元素值:left++;
- 如果left指向的元素值等于待删除的元素值:直接将right指向的元素赋值给left指向的元素,然后right--;
- 重复1和2,直到left > right结束。
代码
class Solution {
public int removeElement(int[] nums, int val) {
//判断数组是否合法
if(nums == null || nums.length == 0) return 0;
int left = 0, right = nums.length - 1;
while(left <= right){
if(nums[left] == val){
nums[left] = nums[right];
right--;
}else{
left++;
}
}
return left;
}
}