【每日一题】Leetcode 15

15. 三数之和

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

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

示例

给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

解题思路

参考官方题解。最暴力的解法是三层嵌套依次列举三个数的和，找到所有满足条件的组合。时间复杂度可想而知是O(n³)，超时不要不要的。
然后这里有一种设想：我们可以不可以实现类似于称重的感觉——超重了就退掉一点；不足再加一些。但是我们的数组是杂乱无序的，因此，给它们排序成了必要一步。nums.sort()走起。
排序之后，从数组最左侧起选择一个数字依次作为a，然后在剩下的数字里找b和c。这其中嵌套一层选次小数b的循环，c我们也从数组中最右侧，也就是最大的数字开始尝试。如果三数之和大于0，则优先左移c，直到a+b+c不大于0（注意b == c时也就不用再继续试，不满足题目中不能重复使用的条件）。
c移动结束后，再次右移b；b结束后才移动a。整个算法试排序算法用去O(nlogn)，两层嵌套是O(n²)。python的提交执行时间是1352ms，挺慢的，该种方法的唯一好处就是直观好想了。上代码：

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

        for a in range(n):
            if a > 0 and nums[a] == nums[a - 1]:
                continue
            c = n - 1
            for b in range(a+1, n):
                if b > a + 1 and nums[b] == nums[b - 1]:
                    continue
                while b < c and nums[a] + nums[b] + nums[c]  > 0:
                    c -= 1
                if b == c:
                    break
                if nums[a] + nums[b] + nums[c] == 0:
                    ans.append([nums[a], nums[b], nums[c]])

        return ans