LeetCode：15. 三数之和

LeetCode：15. 三数之和

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

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

示例：

给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

LeetCode链接

思路：

• 排序
• 从头遍历，固定第 i 个元素，双指针从 i+1、len(nums)-1 处向中间遍历，寻找和为 0 的组合
• 优化：
  • 遍历过程中，跳过重复元素

附代码（Python3）：

class Solution:
    def threeSum(self, nums):
        nums.sort()                          # 排序
        res = []
        for i in range(len(nums)-2):        # 头部遍历
            if i>0 and nums[i]==nums[i-1]:  # 跳过重复元素
                continue
            left, right = i+1, len(nums)-1
            while left < right:              # 双指针遍历
                if nums[left]+nums[right] == -nums[i]:     
                    res.append([nums[i], nums[left], nums[right]])            # 满足条件添加组合
                    while left+1 < len(nums) and nums[left]==nums[left+1]:  # left跳过重复元素
                        left += 1
                    left += 1
                    while right-1 > 0 and nums[right]==nums[right-1]:       # right跳过重复元素
                        right -= 1
                    right -= 1
                elif nums[left]+nums[right] > -nums[i]:
                    right -= 1
                else:
                    left += 1
        return res

test = Solution()
nums_li = [[-1, 0, 1, 2, -1, -4], [0,0,0]]
for nums in nums_li:
    print(test.threeSum(nums))

[[-1, -1, 2], [-1, 0, 1]]
[[0, 0, 0]]