[算法] LeetCode 15.三数之和

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了[算法] LeetCode 15.三数之和，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1651字，纯文字阅读大概需要3分钟。

内容图文

题目：

解法1：当做“两数之和”题目的续集

前情提要传送门：[算法] LeetCode 1.两数之和（python）

把 -c 当做上一题的target，然后用上次的哈希方法找到时a+b=-c成立的a和b。
难点在于输出全部结果的同时不能重复，所以先给数组排序很关键。这个算法三个地方保证了a, b, c三个数字不同时重复，见注释。
时间复杂度O(n2)

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort() #先排序，方便解决“不能重复”这一问题
        n = len(nums)
        result = []
        for j in range(n):
            if j>0 and nums[j]==nums[j-1]:#使a无重复
                continue
            target = -nums[j]       #把第三个数的相反数当做“两数之和”题目里的target
            d = {}
            l = []      #用来解决b的重复问题：屏蔽掉字典（hash表）里被用过的元素
            for i in range(j+1,n):
                a = target - nums[i]
                if nums[i] in d:
                    if nums[i] in l:
                        continue
                    result.append([ nums[j], nums[d[nums[i]]], nums[i] ])
                    l.append(nums[i])
                    while i<n-1 and nums[i]==nums[i+1]: #使c无重复
                        i+=1
                else:
                    d[a] = i

        return result

解法2：双指针

这是从LeetCode上看到别人的解法，如果没有做过两数之和题目，下面这种算法可能更直接。
还是一样，先排序，外循环定住三个数字之一。然后就是在找其他两个数字时用双指针了。因为之前排过序，所以指针移动时会有很方便。
时间复杂度O(n2)

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums.sort() #方便完成不重复的任务，也便于决定两个指针如何移动
        n = len(nums)
        result = []
        for i in range(n):
            if i>0 and nums[i]==nums[i-1]:
                continue
            left = i+1
            right = n-1
            while right>left:
                cur_sum = nums[i]+nums[left]+nums[right]
                if cur_sum == 0:
                    result.append([nums[i], nums[left], nums[right]])
                    while right>left and nums[left] == nums[left+1]:
                        left+=1
                    while right>left and nums[right] == nums[right-1]:
                        right-=1
                    left+=1
                    right-=1
                elif cur_sum > 0:
                    right-=1
                else:
                    left+=1
        return result

内容总结

以上是互联网集市为您收集整理的[算法] LeetCode 15.三数之和全部内容，希望文章能够帮你解决[算法] LeetCode 15.三数之和所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端

---