这篇文章主要介绍LeetCode如何解决三数之和问题,文中介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们一定要看完!
1
题目描述
给定一个整数数组nums,判断nums中是否存在三个元素 a,b,c ,使得 a + b + c = 0 。如不存在返回[],如存在返回所有满足条件且不重复的答案。如:输入[-1,0,1,2,-1,-4]返回[[-1,0,-1],[-1,-1,2]],如输入[-3,3],返回[]。
2
解题
思路一:哈希表
本题要找到满足条件的三个元素,当固定第一个元素a,则题目转化成找到b、c使得和为-a的问题,即与LeetCode刷题DAY 8:两数之和中问题一致,因此也可用哈希表的方法解决。
class Solution: def threeSum(self, nums: List[int]) -> List[List[int]]: if len(nums)<3: return [] nums = sorted(nums) a = list() for i in range(len(nums)-2): if i>0 and nums[i]==nums[i-1]: continue h_map = {} target = -nums[i] for j in range(i+1,len(nums)): if target - nums[j] in h_map: a.append(sorted([nums[i],nums[j],target-nums[j]])) h_map[nums[j]]=j return list(set([tuple(t) for t in a]))
思路二:双指针
当对数组完成排序并固定第一个元素a,则题目与LeetCode刷题DAY 9:两数之和II中问题一致,可用双指针方法解决。
class Solution: def threeSum(self, nums: List[int]) -> List[List[int]]: if len(nums)<3: return [] nums = sorted(nums) a = list() for i in range(len(nums)-2): if i>0 and nums[i]==nums[i-1]: continue x = i+1 y = len(nums)-1 target = -nums[i] while x<y: if nums[x]+nums[y] == target: a.append(sorted([nums[i],nums[x],nums[y]])) x += 1 elif nums[x]+nums[y] < target : x += 1 else: y -= 1 return list(set([tuple(t) for t in a])
以上是“LeetCode如何解决三数之和问题”这篇文章的所有内容,感谢各位的阅读!希望分享的内容对大家有帮助,更多相关知识,欢迎关注亿速云行业资讯频道!
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。