题目描述:给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
解法一(my solution)
def threeSum(self, nums: List[int]) -> List[List[int]]:
mylist = []
length = len(nums)
if(length<3):
return []
for i in range(length-2):
for j in range(i+1,length-1):
cuur = 0-nums[j]-nums[i]
if(cuur in nums[j+1:]):
cuurlist = sorted([nums[i],nums[j],cuur])
if(cuurlist not in mylist):
mylist.append(cuurlist)
return mylist
解法二
def threeSum(self, nums: List[int]) -> List[List[int]]:
nums.sort()
res = []
i = 0
for i in range(len(nums)):
if i == 0 or nums[i]>nums[i-1]:
l = i+1
r = len(nums)-1
while l < r:
s = nums[i] + nums[l] +nums[r]
if s ==0:
res.append([nums[i],nums[l],nums[r]])
l +=1
r -=1
while l < r and nums[l] == nums[l-1]:
l += 1
while r > l and nums[r] == nums[r+1]:
r -= 1
elif s>0:
r -=1
else :
l +=1
return res
- 先将数组进行排序
- 从左侧开始,选定一个值为定值 ,右侧进行求解,获取与其相加为 00的两个值
类似于快排,定义首和尾 - 首尾与 定值 相加
等于 00,记录这三个值
小于 00,首部右移
大于 00,尾部左移
定值右移,重复该步骤
作者:githber
链接:https://leetcode-cn.com/problems/3sum/solution/three-sum-giftu-jie-by-githber/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。