题解 | 三数之和

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param num int整型一维数组 
# @return int整型二维数组
#
class Solution:
    def threeSum(self , num: List[int]) -> List[List[int]]:
        # write code here 固定1位寻找其余两位，用集合去重，集合元素必须为不可变元素
        num.sort()
        n, ans = len(num), set()
        if n<3 or num[0]>0 or num[-1]<0:
            return []
        for i in range(n-2):
            l, r = i+1, n-1
            if num[i]+num[-1]+num[-2]<0:
                continue
            while l<r:
                s = num[i]+num[l]+num[r]
                if s<0:
                    l += 1
                if s>0:
                    r -= 1
                if s==0:
                    ans.add((num[i],num[l],num[r]))
                    l += 1
        return sorted(list(ans),key=lambda x:x)