三元组相加获得target

三元组相加获得target

3Sum

• 给定一个数组，选择三个元素相加，结果为target，找出所有符合的三元组
• Given an array S of n integers, are there elements a, b, c in S such that a + b + c = 0
• Find all unique triplets in the array which gives the sum of zero.

Note: The solution set must not contain duplicate triplets.

For example, given array S = [-1, 0, 1, 2, -1, -4]

example 1

input: [-1, 0, 1, 2, -1, -4]
A solution set is:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

思路

1. 乱序数组，需要找到所有组合，需要三层循环，复杂度为O(n³)。
2. 可以先排序，排序后只需要两层循环，复杂度为O(n²)。第一层循环遍历所有元素，第二层循环由于数组已经排序，只需要头尾两个指针像中间靠拢，一但三个元素相加为target，则添加此三元组，然后继续像中间靠拢扫描。
3. 需要避免重复的三元组被加入

代码

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        ret = []
        for i in range(len(nums) - 2):
            # 避免重复
            if i > 0 and nums[i] == nums[i-1]:
                continue
            j, k = i + 1, len(nums) - 1
            while j < k:
                if nums[i] + nums[j] + nums[k] == 0:
                    ret.append([nums[i], nums[j], nums[k]])
                    j += 1
                    k -= 1
                    # 避免重复
                    while j < k and nums[j] == nums[j - 1]:
                        j += 1
                    while j < k and nums[k] == nums[k + 1]:
                        k -= 1
                elif nums[i] + nums[j] + nums[k] < 0:
                    j += 1
                else:
                    k -= 1
        return ret

本题以及其它leetcode题目代码github地址: github地址