【leetcode15】三数之和-python实现

问题描述：
给你一个包含 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]]

双指针

双指针的思想就是，每次固定一个数，设置左右两个指针，对三数之和的情况进行一个判断。

步骤：

• 首先对数组进行排序
• 遍历排序数组：
  • 当i > 0 时，如果nums[i] == nums[i-1]，那么跳过
  • 设置两个指针：left = i + 1和right = len(nums) - 1，当left < right时，执行循环
  • 当 nums[i] + nums[left] + nums[right] < 0，说明nums[left]小了，left右移
  • 当 nums[i] + nums[left] + nums[right] > 0，说明nums[right]大了，right左移
  • 当 nums[i] + nums[left] + nums[right] == 0，执行循环
    • 判断左界和右界是否和下一位置重复，去除重复解
    • 并同时将 left和right 移到下一位置，寻找新的解

def threeSum(nums):
    # 先排序
    nums.sort()
    res = []

    for i in range(len(nums)-2):
        # 如果当前的数等于前面的数，continue
        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]
            # 如果三数之和小于0，那么左指针右移
            if sum < 0:
                left += 1
            # 如果三数之和大于0，那么右指针左移
            if sum > 0:
                right -= 1
            # 如果三数之和等于0
            if sum == 0:
                # 把三个数添加到结果中
                res.append([nums[i], nums[left], nums[right]])
                # 去除重复的左边元素
                while (left < right) and (nums[left] == nums[left+1]):
                    left += 1
                # 去除重复的右边元素
                while (left < right) and (nums[right] == nums[right-1]):
                    right -= 1
                # 更新指针
                left += 1
                right -= 1

    return res



nums = [-1,0,1,2,-1,-4]
print(threeSum(nums))