机试打卡 -08 三数之和（排序&双指针）

---

 若直接选择在原数组nums中，采用三重循环确定答案，时间复杂度为O(N**3)，超时……

官方题解：排序 + 双指针

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        n = len(nums)

        # 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