leetcode-15 三数之和

leetcode-15 三数之和

参考：负雪明烛

题目描述：

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

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort()
        N = len(nums)
        res = []
        for t in range(N-2):
            if t>0 and nums[t] == nums[t-1]:
                continue
            i,j=t+1,N-1
            while i<j:
                sum_ = nums[t] + nums[i] + nums[j]
                if sum_ == 0:
                    res.append([nums[t],nums[i],nums[j]])
                    i += 1
                    j -= 1
                    while i<j and nums[i] == nums[i-1]:
                        i += 1
                    while i<j and nums[j] == nums[j+1]:
                        j -= 1
                elif sum_ < 0:
                    i += 1
                else:
                    j -= 1
        return res