目录
介绍
给你一个整数数组 nums
,判断是否存在三元组 [nums[i], nums[j], nums[k]]
满足 i != j
、i != k
且 j != k
,同时还满足 nums[i] + nums[j] + nums[k] == 0
。请
你返回所有和为 0
且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]] 解释: nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。 nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。 nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。 不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。 注意,输出的顺序和三元组的顺序并不重要。
示例 2:
输入:nums = [0,1,1] 输出:[] 解释:唯一可能的三元组和不为 0 。
示例 3:
输入:nums = [0,0,0] 输出:[[0,0,0]] 解释:唯一可能的三元组和为 0 。
解题思路
解决三数之和问题的一种高效方法是使用双指针法与排序。该方法基于以下观察:要求三个数的和为 0,可以转化为在排序后的数组中寻找两个数,使得它们的和等于一个固定的目标值。具体步骤如下:
- 对数组进行排序,以便后续操作。
- 遍历排序后的数组,固定一个数
nums[i]
,并使用双指针left
和right
分别指向i
后面的起始位置和末尾位置。 - 在
left
和right
指针范围内,通过计算nums[i] + nums[left] + nums[right]
的和来判断是否满足条件。如果和为 0,则将这个三元组[nums[i], nums[left], nums[right]]
添加到结果中。 - 如果和小于 0,则将
left
指针右移一位,以增大和;如果和大于 0,则将right
指针左移一位,以减小和。 - 在移动指针时,要注意跳过重复的元素,以避免重复的三元组。
代码实现
def threeSum(nums):
nums.sort()
result = []
for i in range(len(nums) - 2):
# Skip duplicate values for i
if i > 0 and nums[i] == nums[i - 1]:
continue
left, right = i + 1, len(nums) - 1
while left < right:
total = nums[i] + nums[left] + nums[right]
if total == 0:
result.append([nums[i], nums[left], nums[right]])
# Skip duplicate values for left and 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
elif total < 0:
left += 1
else:
right -= 1
return result
总结
双指针法与排序是解决三数之和问题的一种高效方法,其时间复杂度为 O(n^2),其中 n 是数组的长度。通过对数组排序,并使用双指针遍历数组,我们可以在一次遍历中找到所有和为 0 且不重复的三元组,从而解决这个问题。希望本篇博客能帮助你更好地理解和应用双指针法与排序解决类似的问题。