三数之和

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

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

和平方数之和的矩阵解法相似，只不过那个是二维的，都是这个是三维的。

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        l = len(nums)
        if l<3:
            return []
        res = []
        nums.sort()
        for i in range(l-2):
            if i>0 and nums[i] == nums[i-1]:continue # 去除重复遍历，保证不会生成重复三元组
            j, k = i+1, l-1
            while j<k: # 只遍历下三角元素
                s = nums[i] + nums[j] + nums[k]
                if s<0:
                    j += 1 # 和小于0，则需要增大和，j+1使得num[j]增大
                    while j<k and nums[j]==nums[j-1]:j+=1
                elif s>0: # 和大于0，则需要减小和，k-1使得num[k]减小
                    k -= 1
                    while j<k and nums[k]==nums[k+1]:k-=1
                else: # 和等于0
                    res.append([nums[i], nums[j], nums[k]])
                    j += 1
                    while j<k and nums[j]==nums[j-1]:j+=1
                    k -= 1
                    while j<k and nums[k]==nums[k+1]:k-=1
        return res

然后需要一些细节处理，解题的时候注意就行。