【LeetCode】15. 3Sum

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.

题意：找出数组中所有和为0的不同的三个数，从小到大排列

思路：

刚开始想用双指针，然后在两个指针中间去找第三个数，但是想不到边上的两个指针往中间递进的条件。

然后就只有先确定一个数，然后用双指针去找另外两个数

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        l = len(nums)
        if l<3:
            return []
        res = []
        nums.sort()
        i=0
        while(i<l-2):
            if nums[i]>0:
                break;
            target = 0 - nums[i]
            start = i+1;end = l-1
            while start<end:
                tmp = nums[start] + nums[end]
                if tmp>target:
                    end-=1
                elif tmp<target:
                    start+=1
                else:
                    res.append([nums[i],nums[start],nums[end]])
                    t = start
                    while start<end and nums[t]==nums[start]:　　　　　　#略过相同的数
                        start+=1
                    t = end
                    while start<end and nums[t]==nums[end]:　　　　　　　　#略过相同的数
                        end-=1
            t=i
            while i<l-2 and nums[t]==nums[i]:
                i+=1
        return res

 .。。终于想到了，可以用双指针，用二分法找第三个值，但是要用到递归

。。。但是这种方法没ac，因为递归生成输出的结果。。。。有重复，这个题够折磨了，先到这吧

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        l = len(nums)
        if l<3:
            return []
        res = []
        nums.sort()
        start=0;end=l-1
        flag=0
        while start<end-1:
            tmp = nums[start]+nums[end]
            target = 0-tmp
            mid = (start+end)//2
            tstart=start;tend=end
            while tstart<mid<tend and nums[tstart]<=target<=nums[tend]:
                if nums[mid]==target:
                    flag=1
                    res.append([nums[start],nums[mid],nums[end]])
                    break
                elif nums[mid]>target:
                    t=mid
                    mid=(tstart+mid)//2
                    tend=t
                else:
                    t = mid
                    mid=(mid+tend)//2
                    tstart=t
            if flag==1:
                flag=0
                
                subnums=self.threeSum(nums[start:end])
                if subnums!=[]:
                    for x in subnums:
                        res.append(x)
                
                subnums=self.threeSum(nums[start+1:end+1])
                if subnums!=[]:
                    for x in subnums:
                        res.append(x)
                            
                t=start
                while start<end and nums[t]==nums[start]:
                    start+=1
                t=end
                while start<end and nums[t]==nums[end]:
                    end-=1
            else:
                if tmp<0:
                    start+=1
                else:
                    end-=1
        
        return res