【LeetCode】每日一题：三数之和

解题思路

最开始是打算沿着二数之和的思路做，即固定了最大的，然后小的开始遍历，因为这种遍历方式只需要遍历一轮就能完成，所以复杂度应该是O（n2），但是最后几个示例还是超时了，可能进一步优化还是可以做的。但是事实上，我最开始就在寻找一个能够约束剩余两个变量的方法，因为不重复我们可以添加大小关系的假设，但是脑子里一直是两个指针在一个遍历中移动所以没搞出来。事实上是，一个变量变大，另一个变量的上限减小，可以基于这个想法做。

未AC代码，最后样例超时

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        def get_res(index, k):
            disk = {}
            res = []
            for i, n in enumerate(nums[:index]):
                if n <= - k // 2:
                    if n == -k // 2 and n in disk and n + n + k == 0:
                        res.append([n, n, k])    
                    disk[-k - n] = i
                else:
                    if n in disk:
                        res.append([n, - n - k, k])
            return res
                
        res = []
        nums.sort()
        for i in range(-1, -len(nums) - 1, -1):
            temp = get_res(i, nums[i])
            for t in temp:
                if t not in res:
                    res.append(t)
        return res

官方题解

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        n = len(nums)
        nums.sort()
        ans = list()
        
        # 枚举 a
        for first in range(n):
            # 需要和上一次枚举的数不相同
            if first > 0 and nums[first] == nums[first - 1]:
                continue
            # c 对应的指针初始指向数组的最右端
            third = n - 1
            target = -nums[first]
            # 枚举 b
            for second in range(first + 1, n):
                # 需要和上一次枚举的数不相同
                if second > first + 1 and nums[second] == nums[second - 1]:
                    continue
                # 需要保证 b 的指针在 c 的指针的左侧
                while second < third and nums[second] + nums[third] > target:
                    third -= 1
                # 如果指针重合，随着 b 后续的增加
                # 就不会有满足 a+b+c=0 并且 b<c 的 c 了，可以退出循环
                if second == third:
                    break
                if nums[second] + nums[third] == target:
                    ans.append([nums[first], nums[second], nums[third]])
        
        return ans