LeetCode-15. 三数之和

题目描述

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

思路1

      最后去重

1. 先将数组排序
2. 从左到右依次固定第k个数
3. 然后在固定数后面的区间里，找到两个数，使它们的和等于负固定的数（按照有序数组中找一个和为target的两个数的方法来做，双指针）
4. 循环结束条件：当固定的数<len(nums)-2 and nums[k]<=0 (因为一旦大于0，后面两个数就拼凑不出等于0的数了)

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        if len(nums)<3:
            return []
        nums.sort()
        k=0
        res=[]
        while k<len(nums)-2 and nums[k]<=0:
            target=-nums[k]
            i=k+1
            j=len(nums)-1
            #应对测试用例中全是大量的0的情况
            if nums[k]==nums[j]:
                j=k+2
            while i<j:
                sumnum=nums[i]+nums[j]
                if sumnum>target:
                    j-=1
                elif sumnum<target:
                    i+=1
                else:
                    res.append([nums[k],nums[i],nums[j]])
                    i+=1
                    j-=1
            k+=1
        return list(set([tuple(x) for x in res])) # 二维数组去重的方法

思路2

     边搜索边去重

1. 固定一个数，找另外两个数，大体思路一样
2. 就是判断一下，当前固定的数是否等于前一个数，如果等于就直接固定下一个
3. 在搜索另外两个数也一样，如果当前数等于前一个数，也直接跳过（避免重复）

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        if len(nums)<3:
            return []
        nums.sort()
        k=0
        res=[]
        while k<len(nums)-2 and nums[k]<=0:
            if k>0 and nums[k]==nums[k-1]:
                k+=1
                continue
            i=k+1
            j=len(nums)-1
            target=-nums[k]
            while i<j:
                sumnum=nums[i]+nums[j]
                if sumnum>target:
                    j-=1
                    while i<j and nums[j]==nums[j+1]:
                        j-=1
                elif sumnum<target:
                    i+=1
                    while i<j and nums[i]==nums[i-1]:
                        i+=1
                else:
                    res.append([nums[k],nums[i],nums[j]])
                    i+=1
                    j-=1
                    while i<j and nums[i]==nums[i-1]:
                        i+=1
                    while i<j and nums[j]==nums[j+1]:
                        j-=1
            k+=1
        return res