15. 3Sum

【题目】

Given an array nums of n integers, are there elements a, b, c in nums such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.

【注意】

The solution set must not contain duplicate triplets.

【举例】

Given array nums = [-1, 0, 1, 2, -1, -4],

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

【题目翻译】

大意就是从一个数组里面找出所有和为0的三元组， 不能包括重复的， 举例如上。

【解题思路】

这题乍看可以用回溯法解决，

先选前三个数[-1,0,1]判断， 然后把第三个数去掉换成第四个数[-1,0,2]再判断, 然后把第四个数去掉换成第五个数[-1,0,-1]再判断...

判断完-1, 0 为头两个位置的数后， 再判断-1， 1为头两个位置的情况， 再判单-1， 2为头两个位置的情况...

判断为-1为起始位置的所有情况以后， 再判断0为起始位置的所有情况...

【回溯代码】

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        result = []
        self.backtracking(result, [], 0, nums)
        return result

    
    def backtracking(self, result, tmp, start, nums):
        if len(tmp) == 3:
            if sum(tmp) == 0:
                result.append(list(tmp))
            return
        
        for i in range(start, len(nums)):
            if len(tmp) < 3:
                tmp.append(nums[i])
                self.backtracking(result, tmp, i+1, nums)
                tmp.pop()

输出结果是：

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

发现回溯法解决不了数组重复的问题。

 

 

其实这道题还有另一种解法：

首先对数组nums进行升序排序， 然后遍历所有三元组可能的起始元素nums[0], nums[len(nums)-3],  假设遍历到了起始元素nusm[i], 然后再剩余元素里面找两个和为-nums[i]的元素， 与起始元素组成和为零的三元组。

如果起始元素nums[i] == nums[i-1], nums[i]为起始元素的可能就不用再考虑， 因为所有可能已经包含在了nums[i-1]为起始元素的情况里。

【代码】

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        res = []
        for i in range(len(nums)-2):
            if i == 0 or nums[i] != nums[i-1]: #如果该起始元素等于上一个起始元素，跳过
                #在剩余元素里找和为-nums[i]的组合
                lo = i + 1
                hi = len(nums) - 1
                sum_ = -nums[i]
                while lo < hi:
                    if nums[lo] + nums[hi] == sum_:
                        res.append([nums[i], nums[lo], nums[hi]])
                        while lo < hi and nums[lo+1] == nums[lo]: lo += 1 #避免重复
                        while lo < hi and nums[hi-1] == nums[hi]: hi -= 1
                        hi -= 1
                        lo += 1
                    elif nums[lo] + nums[hi] > sum_:
                        hi -= 1
                    else:
                        lo += 1
        return res