15. 3Sum

题目链接：https://leetcode.com/problems/3sum/

代码

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        count={}
        #统计出现的次数，放入dict里面，其中key为值，value为次数
        for num in nums:
            if num not in count:
                count[num]=1
            else:
                count[num]+=1
        #对key进行排序
        nums=sorted(count)
        res=[]
        for i,n in enumerate(nums):
            if count[n] >= 2 :
                if n==0 :
                    if count[n]>=3 :
                        res.append([0,0,0])
                else: 
                    if -2*n in count:
                        res.append([n,n,-2*n])
            if n<0:
                twosum=-n
                #缩小查询范围
                left=bisect.bisect_left(nums,twosum-nums[-1],i+1)
                right=bisect.bisect_right(nums,twosum//2,left)
                for j in nums[left:right]:
                    comp = twosum - j
                    if comp in count and comp != j:
                        res.append([n,j,comp])
        return res

思路详解

• 统计每个值出现的次数放入dict里面，其中key为数值，value为出现的次数

• 对key进行排序，返回一个列表nums

• 遍历nums

  • 如果次数大于等于2的情况

    • 值为0，次数大于2，那么添加一项（0,0,0）
    • 值n!=0，次数为2，那么若-2n在字典中，那么添加一项（n,n,-2n）

    其他情况都不满足

  • 因为target为0，所以可以只考虑负值情况，转为对二值进行处理的情况。其中，有个缩小查询范围的操作，求两者之和为twosum，那么较小的那个值范围在[twosum-nums[-1],twosum//2]之间
    
    #缩小查询范围
    left=bisect.bisect_left(nums,twosum-nums[-1],i+1)
    right=bisect.bisect_right(nums,twosum//2,left)

  关于bisect模块可参考https://blog.csdn.net/SummerHmh/article/details/89918164