三数之和-双指针

题目描述

给一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。

注意： 答案中不可以包含重复的三元组。

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为： [ [-1, 0, 1], [-1, -1, 2] ]

解题思路

指定一个相对不动的的指针i，两个左右指针分别指向i的下一个位置和数组的最后一个位置。当指针i不动时，对左右指针进行操作。那么操作的原则是什么呢？首先要将nums数组中的元素进行大小排序，sort之后从左到右递增，i首先指向首位0，此时左指针left指向1，右指针指向len(nums)-1。

1. 当nums[i]>0时，显然最小值大于0，这个数组中没有满足题目要求的三元组。
2. 计算sum(nums[i]+nums[left]+nums[right])=sums

• sums>0;为了减小sums，right-=1
• sums<0;为了增大sums,   left+=1
• sums=0；得到一个满足条件的三元组，此时在i不变的情况下，继续移动left和right，btw考虑到数组中有重复元素，需要对左右指针指向的值进行判定，遇到重复值时，加速指针的移动。

以上为相对不动指针指向i时的一次判断过程。后续i需要遍历整个数组

显然，不仅左右指针会遇到重复值，不动指针i也会遇到，当i遇到重复值时，跳出当前循环判断，i指向下一不同值的位置继续进行后续判断。

代码实现

class Solution:
    def threeSum(self, nums):
        array = []
        L = len(nums)
        nums.sort()
        for i in range(L):
            if nums[i]>0:
                break
            elif i>0 and nums[i] == nums[i-1]：
                continues
            left = i + 1
            right = L - 1
            while left < right:
                total = nums[i]+nums[left]+nums[right]
                if total > 0:
                    right-=1
                elif total < 0:
                    left+=1
                else:
                    array.append([nums[i],nums[left],nums[right]])
                    while left!=right and nums[left]==nums[left+1]:left+=1
                    while left!=right and nums[right]==nums[right-1]:right-=1
                    left+=1
                    right-=1
        return array