LeetCode: 三数之和 python实现

15. 三数之和

1. 题目描述

给你一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？请你找出所有满足条件且不重复的三元组。
注意：答案中不可以包含重复的三元组。
示例：
给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

2. 代码实现

在主要思想：
用三个指针k ：数指针 , l ：左指针, r ：右指针,
指针 l 从左往右移动，且始终比 k 大一位，这样就保证不会跟 k 重叠，
指针 r 从右往左移动，且始终比 l 大，这样 l 和 r 就不会重叠，即三个指针都不会重叠。
在这个基础上，实现主要逻辑：
先将数字排序

1. nums[k]>0时，可以直接跳出循环，因为nums[k]都比0大了，后面的nums[i]和nums[j]肯定更大，三者加起来肯定大于0
2. nums[k]和nums[k-1]相等，前后元素重复，需要过滤
3. 如果nums[l]+nums[r]+nums[k]>0，即三者之和太大了，将 r 指针左移
4. 如果nums[l]+nums[r]+nums[k]<0，说明三者之和太小了，同理将 l 指针右移
5. 如果nums[l]+nums[r]+nums[k]==0，这就是要找的答案，将其保存起来，同时 l 右移，r左移
   注意： l 和 r 在移动的过程中还需要判断前后元素是否重复

def threeSum(nums):
    """
    :type nums: List[int]
    :rtype: List[List[int]]
    """
    nums.sort()
    lens = len(nums)
    res = []
    for i in range(lens-2):
        k = i
        if nums[k] > 0: break
        if k>0 and nums[k-1]==nums[k]:continue
        l, r = k+1, lens-1
        while l<r:
            if nums[k]+nums[l]+nums[r]==0 :
                res = res + [[nums[k], nums[l], nums[r]]]
                l += 1
                r -= 1
                while l<r and nums[l]==nums[l-1]:
                    l += 1
                while l<r and nums[r]==nums[r+1]:
                    r -= 1
            elif nums[k]+nums[l]+nums[r]>0:
                r -= 1
                while l<r and nums[r] == nums[r + 1]:
                    r -= 1
            else:
                l += 1
                while l<r and nums[l] == nums[l - 1]:
                    l += 1
    return print(res)

测试：

threeSum([-1, 0, 1, 2, -1, -4])
threeSum([-2,0,0,2,2])
threeSum([0,0,0])

输出：

[[-1, -1, 2], [-1, 0, 1]]
[[-2, 0, 2]]
[[0, 0, 0]]