算法通关村——双指针的妙用

1 题目：移除元素

给你一个数组 nums 和一个值 val，你需要 原地 移除所有数值等于 val 的元素，并返回移除后数组的新长度。

不要使用额外的数组空间，你必须仅使用 $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 中唯一元素的个数。

题目来源：LeetCode26.删除有序数组中的重复项

思想：定义两个指针 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.删除有序数组中的重复项这两道问题。双指针法是一种高效、简单且不需要额外空间的算法技巧。双指针法通常只需要在原始数组上进行操作，而不需要使用额外的数据结构，因此节省了空间复杂度，而且双指针法通常在一次遍历中完成问题的解决，无需多次遍历数组，效率较高。