三数之和用python来写一个

以下是一个Python实现的三数之和函数：

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

该函数接受一个整数列表作为输入，返回一个列表，其中包含所有三元组，它们的和等于零。该函数首先对列表进行排序，然后通过遍历每个元素i（从0到n-2），在i+1到n-1的范围内使用两个指针l和r来查找与i组合形成和为零的元素。如果找到这样的三元组，则将其添加到结果列表中。该函数还包含一些优化，例如跳过重复元素和提前退出循环等。