腾讯6-三数之和

原创于 2019-12-20 00:34:52 发布 · 75 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

不完整刷题专栏收录该内容

51 篇文章

订阅专栏

探讨腾讯面试题“三数之和”解决方案，对比“两数之和”，利用排序和二分查找优化算法，避免重复结果，适用于LeetCode15题目。

腾讯6-三数之和leetcode15

列出三数之和为0的所有可能，
和之前腾讯的两数之和不一样，
之前腾讯的两数之和需要返回下标，导致了不能对输入进行排序，因为排序会破坏下标的顺序。
这里不需要返回下标，只列出满足和为0的三个元素，
所以可以利用排序+二分查找了。比较好写

重点理解和之前腾讯的两数之和(取下标，用hash,第一次存位置，第二次取结果)的区别就方便记忆这两道题了。

class Solution:
    def twoSum(self, nums,target,res):
        i,j=0,len(nums)-1
        while i<j:
            if i==0 or nums[i]!=nums[i-1]:
                if nums[i]+nums[j]==target:
                    res.append([-target,nums[i],nums[j]])
                    i+=1
                elif nums[i]+nums[j]<target:
                    i+=1
                else:
                    j-=1
            else:
                i+=1
                
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        res=[]
        nums=sorted(nums)
        for i in range(len(nums)):
            if i==0 or nums[i]!=nums[i-1]:
                self.twoSum(nums[i+1:len(nums)],-nums[i],res)
            else:
                continue
        return res