Leetcode: 三数之和

Leetcode: 三数之和

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

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

先排序，然后从两端开始向中间搜索，i为遍历后的数字，R表示最右端数字，L表示nums[i + 1]，每次遍历后如果三个数字的和为0，就R和L都同时移到下一个索引（R–，L++），这是因为要求去重，如果只改变一个还符合条件的话一定是重复的。因为列表是按顺序排的，所以和>0, 则说明当前的R偏大，因此R-=1； 反之, 说明当前的L偏小，则把L+=1

class Solution:
    def threeSum(self, nums):
        nums.sort()
        res_list = []
        for i in range(len(nums)):
            if i == 0 or nums[i] > nums[i-1]:
                L = i + 1
                R = len(nums) - 1
                while L < R:
                    s = nums[i] + nums[R] + nums[L]
                    if s == 0:
                        res_list.append([nums[i], nums[R], nums[L]])
                        L += 1
                        R -= 1
                        while L < R and nums[L] == nums[L-1]:
                            L += 1
                        while R > L and nums[R] == nums[R+1]:
                            R -= 1
                    elif s > 0:
                        R -= 1
                    else:
                        L += 1
        return res_list