双指针法小结 --以LeetCode 27、977、15、18为例

初刷力扣，还是先从基础的做起，而双指针法在基础题中的应用非常广泛。

LeetCode 27、977、15、18就是我这几天所做到的可以用双指针法很好解决的题目：

27：移除元素

相当于把数组分为两部分：前一部分为有效部分，不等于val的数，后一部分为无效部分，即等于val的数，最后返回前一部分的长度，直接上代码：

class Solution {
    public int removeElement(int[] nums, int val) {
        int j = nums.length - 1;
        for (int i = 0; i <= j; i++) {
            if (nums[i] == val) {
                swap(nums, i--, j--);
            }
        }
        return j + 1;
    }
    void swap(int[] nums, int i, int j) {
        int tmp = nums[i];
        nums[i] = nums[j];
        nums[j] = tmp;
    }
}

题干的说明也要稍稍注意一下:

977：有序数组的平方

 因为给定数组为非递增顺序，且返回平方后的数组也要是非递增顺序，即平方后的最大值要么在参数数组的头部和尾部。此时即可用双指针的思想，在数组头部和尾部各放置一个指针进行操作。

《代码随想录》中有一张一目了然的动图：

class Solution {
    public int[] sortedSquares(int[] nums) {
        int r=nums.length-1;
        int l=0;
        int []res=new int[r+1];
        int index=res.length-1;//这里不能令index=0
        //因为只能确定平方后最大的数在两端，所以必须这样倒着将结果放入
        while(l<=r)
        {
            if(nums[l]*nums[l] > nums[r]*nums[r])
            {
                res[index--]=nums[l]*nums[l];
                l++;
            }
            else{
                res[index--]=nums[r]*nums[r];
                r--;
            }
        }
        return res;
    }
}

15：三数之和

 这题一开始我想用哈希法来着，不过试着试着就放弃了，似乎还是双指针法更简洁：

大致思路是：首先将数组排序，然后for循环，i从下标0的地方开始，同时定一个下标left 定义在i+1的位置上，定义下标right 在数组结尾的位置上。

当nums[left]+nums[right]+nums[i]>0时，说明三数之和过大，right下标就应该向左移动，这样才能让三数之和小一些。<0时同理，移动left下标。

而nums[left]+nums[right]+nums[i]=0时，也并不是万事大吉了，因为不能计入重复出现的数，所以还需要有进一步的优化操作。

class Solution {
    public List<List<Integer>> threeSum(int[] nums) {
        Arrays.sort(nums);
        List <List<Integer>> result= new ArrayList<>();
        for(int i=0;i<nums.length;i++)
        {
            if (i > 0 && nums[i] == nums[i - 1])
                continue;//因为答案不包含重复，所以遇见先前已遍历的数字就跳出本次循环

            int left=i+1;
            int right=nums.length-1;
            while(left<right)
            {
                int sum=nums[left]+nums[right]+nums[i];
                if(sum>0)
                    right--;
                else if(sum<0)
                    left++;
                //当sum为0时
                else{
                    result.add(Arrays.asList(nums[left],nums[right],nums[i]));
                    while (right > left && nums[right] == nums[right - 1]) right--;
                    while (right > left && nums[left] == nums[left + 1]) left++;
                    //以上均是为了防止计入重复的数
                    left++;right--;
                }
            }
        }
        return result;
    }
}

18：四数之和

 这题算是上一题的翻版，运用的思想都是一样的：

三数之和的双指针解法是一层for循环num[i]为确定值，然后循环内有left和right下标作为双指针，找到nums[i] + nums[left] + nums[right] == 0。

四数之和的双指针解法是两层for循环nums[k] + nums[i]为确定值，依然是循环内有left和right下标作为双指针，找出nums[k] + nums[i] + nums[left] + nums[right] == target的情况，三数之和的时间复杂度是O(n2)，四数之和的时间复杂度是O(n3) 。

class Solution {
    public List<List<Integer>> fourSum(int[] nums, int target) {
        Arrays.sort(nums);
        List<List<Integer>> result=new ArrayList<>();
        for(int i=0;i<nums.length;i++)
        {
            if(i>0 &&nums[i]==nums[i-1])
                continue;
            for(int j=i+1;j<nums.length;j++)
            {
                if(j>i+1 &&nums[j]==nums[j-1])
                    continue;
                int left=j+1;
                int right=nums.length-1;
                while(left<right)
                {
                    int sum=nums[i]+nums[j]+nums[left]+nums[right];
                    if(sum>target)
                        right--;
                    else if(sum<target)
                        left++;
                    else{                                
                      result.add(Arrays.asList(nums[left],nums[right],nums[i],nums[j]));
                        while(right>left &&nums[right]==nums[right-1]) right--;
                        while(right>left &&nums[left]==nums[left+1]) left++;
                        left++;right--;
                    }
                }
            }
        }
        return result;
    }
}

参考资料：代码随想录