#15. 三数之和

#15. 三数之和

题目

给定一个数组nums，找出所有三元组[nums[i], nums[j], nums[k]]，使得nums[i] + nums[j] + nums[k] == 0 且 i != j, j != k, k != i。 （找出所有互异且和为零的数组中的三元组）

解答

锁定一–双指针解法

1. 先把array排序。
2. 锁定第一个位置i，后面设立left = i + 1, right = size - 1。
   1. 如果 下一个 i 和之前的 i 对应的值一样，那么下一个i的情况必定被之前的i的包含，因此跳过。
3. 如果nums[i] + nums[left] + nums[right] == 0，则加入此三元组。并把left，right都移动新的对应值的地方。
4. 如果 < 0: 那么增加left， > 0: 减少right

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        result = []
        # sort the array
        nums.sort()
        # two pointers method
        for i in range(len(nums)):
            # deal with 1st repetition scenario: watch the condition order! 不然溢出
            if i != 0 and nums[i] == nums[i - 1]:
                continue
            # initialize left and right pointers
            left = i + 1
            right = len(nums) - 1
            while left < right: 
                if nums[i] + nums[left] + nums[right] == 0:
                    result.append([nums[i],nums[left],nums[right]])
                    # 不规定left < right会导致数组溢出
                    while left < right and nums[left] == nums[left + 1]:
                        left += 1
                    left += 1
                    while left < right and nums[right] == nums[right - 1]:
                        right -= 1
                    right -= 1
                elif nums[i] + nums[left] + nums[right] < 0:
                    left += 1
                else:
                    right -= 1
        return result

收获

1. 本题的精髓在于排序，排序过后就可以用left ++，right – 等方式极大缩短排查时间。
2. 精髓之二在于3个异位数字，视图排布必定是一个在前，两个在后，对应了我们锁定一–双指针解法。