LeetCode LCR：007.三数之和-Python

链接：LCR 007. 三数之和 - 力扣（LeetCode）

题目描述：

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

示例 1：

输入：nums = [-1,0,1,2,-1,-4]
输出：[[-1,-1,2],[-1,0,1]]

解题思路：双指针

一般想到一个元素是否出现过或者是否在一个集合里的时候，会想到用哈希表，但此题的条件是不重复，用哈希存储较为麻烦，所以我们可用双指针法。

1.可以先将数组从小到大排序

2. 要实现  a + b + c = 0  那么我们假设a从a[0]也就是最小的数开始遍历，b为左指针 left 指向a的下一个元素，c为右指针 right 指向数组最后一个元素也就是最大的数，开始遍历。

3.此时相当于 a = nums[i]，b = nums[left]，c = nums[right]。接下来移动left 和right，

• 如果nums[i] + nums[left] + nums[right] > 0 就说明 此时三数之和大了，因为数组是排序后了，所以right下标就应该向左移动，这样才能让三数之和小一些。
• 如果 nums[i] + nums[left] + nums[right] < 0 说明 此时 三数之和小了，left 就向右移动，才能让三数之和大一些，直到left与right相遇为止。

4.要保证三元组不重复，就要保证三元组中每一个元素和另一个三元组不重复

• a是从nums[0]开始，b是从nums[1]开始，所以要想a不重复，应该是nums[i]与nums[i-1]不重复，因为若是nums[i]与nums[i+1]不重复就是a与b不重复，题目没有规定三元组内元素不重复
• b不重复应当nums[left] != nums[i+1]
• c不重复应当nums[right] != nums[right-1],因为right是从后往前遍历

时间复杂度：O(n^2)。

代码：

class Solution(object):
    def threeSum(self, nums):
        # 排序
        nums = sorted(nums)
        # 结果集
        result = []
        for i in range(len(nums)): 
            # 若排序后第一个元素已>0，则直接返回result
            #（文章后对这里有解释）
            if nums[i] > 0:
                return result
            # 给a去重
            if i > 0 and nums[i]==nums[i-1]:
                continue
            # 初始化左右指针
            left = i + 1
            right = len(nums)-1
            while left < right:
                sum = nums[i] + nums[left] + nums[right]
                if sum > 0:
                    right -= 1
                elif sum < 0:
                    left += 1
                else:
                    result.append([nums[i],nums[left],nums[right]])
                    # b去重
                    while left < right and nums[left] == nums[left+1]:
                        left += 1
                    # c去重
                    while left < right and nums[right] == nums[right-1]:
                        right -= 1
                    left += 1
                    right -= 1
        return result

ps:

我在二次刷题时，出现了问题，为什么 i 遍历时，只要nums[ i ] > 0 就直接返回result呢？

     我还是思索了一下想起来了，因为每一次重新遍历 i 时，是重新开始找新的三元组的过程，而 i 作为我们已排序好的数组的三元组的第一个数，肯定不能大于0呀，要是第一个数都大于0，后面的数只会递增，不可能再有和为0的三元组了，所以三元组已找寻完毕，直接返回result。