题目描述:
给你一个整数数组 nums,判断是否存在三个元素 a,b,c,使得 a + b + c = 0。请你找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = [] 输出:[]
示例 3:
输入:nums = [0] 输出:[]
思路分析:
本题解采用双指针的方法。
先将数组排序,然后遍历数组,对于每个数 nums[i],使用双指针 left 和 right 分别表示剩余元素的左右两端,寻找所有满足 nums[i] + nums[left] + nums[right] = 0 的三元组。如果 nums[i] > 0,说明剩余元素都大于 0,无法组成和为 0 的三元组,直接返回结果即可。如果 nums[i] == nums[i-1],说明上一个循环已经计算过以 nums[i-1] 开头的所有三元组,跳过这个元素以避免重复计算。
在内部循环中,我们通过双指针 left 和 right 向中间移动,寻找所有满足 nums[i] + nums[left] + nums[right] = 0 的三元组。如果当前三元组的和小于 0,说明 left 指针指向的元素太小,需要将 left 向右移动;如果当前三元组的和大于 0,说明 right 指针指向的元素太大,需要将 right 向左移动。如果找到了一个满足条件的三元组,记录下来后分别将 left 和 right 移动到与之不同的位置以避免重复计算。
代码实现:
根据上述思路,我们可以使用 Python3 编写如下代码实现:
class Solution:
def threeSum(self, nums: List[int]) -> List[List[int]]:
n = len(nums)
if n < 3:
return []
# 排序
nums.sort()
res = []
for i in range(n):
if nums[i] > 0:
break
if i > 0 and nums[i] == nums[i-1]:
continue
left, right = i+1, n-1
while left < right:
s = nums[i] + nums[left] + nums[right]
if s < 0:
left += 1
elif s > 0:
right -= 1
else:
res.append([nums[i], nums[left], nums[right]])
while left < right and nums[left] == nums[left+1]:
left += 1
while left < right and nums[right] == nums[right-1]:
right -= 1
left += 1
right -= 1
return res
时间复杂度:O(n^2),其中 n 是数组的长度。
空间复杂度:O(logn)。
总结:
通过本篇博客,我们了解了如何寻找所有满足 a + b + c = 0 的不重复的三元组。这个题目是一个比较典型的数组处理和双指针算法问题,解决这个问题能够提高我们对数组处理和双指针算法的理解能力。
同时,我们还讲解了通过将数组排序,使用双指针向中间移动来寻找所有满足条件的三元组的思路和具体实现。这种思路也可以应用到其他类似问题中,比如寻找最接近目标值的三元组等。
希望本篇博客能对您有所帮助,如果您有任何疑问或建议,请在评论区与我们分享。