给你一个整数数组
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 。
1. 题目解析:
题目要求找出给定数组中所有满足三数之和为0的不重复三元组,并将它们返回。
2. 选择使用双指针法的做法:
在上面的代码中,作者选择了使用双指针法来解决这个问题。这种做法的原因如下:
-
时间复杂度较低:双指针法的时间复杂度为 O(n^2),比暴力枚举法的 O(n^3) 要低,可以更快地找到解决方案。
-
空间复杂度较低:双指针法的空间复杂度为 O(1),只需要常数级别的额外空间,不需要额外的数据结构来存储中间结果。
-
代码逻辑简单:使用双指针法可以将问题分解为两个较小的子问题,分别求解左右两边的元素,然后合并结果,代码逻辑相对简单。
3.代码展示:
class Solution { public: vector<vector<int>> threeSum(vector<int>& nums) { vector<vector<int>> result; sort(nums.begin(), nums.end()); int n = nums.size(); for (int i = 0; i < n - 2; ++i) { if (i > 0 && nums[i] == nums[i - 1]) continue; // 跳过重复的元素 int target = -nums[i]; int left = i + 1; int right = n - 1; while (left < right) { int sum = nums[left] + nums[right]; if (sum == target) { result.push_back({nums[i], nums[left], nums[right]}); while (left < right && nums[left] == nums[left + 1]) left++; // 跳过重复的元素 while (left < right && nums[right] == nums[right - 1]) right--; // 跳过重复的元素 left++; right--; } else if (sum < target) { left++; } else { right--; } } } return result; } };
-
首先,对输入数组
nums
进行排序,以便后续处理。 -
然后,使用双指针法遍历数组。外层循环遍历数组元素
nums[i]
,内层循环使用两个指针left
和right
分别指向nums[i]
的后面两个元素。 -
在内层循环中,根据题目要求,判断
nums[left] + nums[right]
是否等于-nums[i]
,如果等于,则找到了一个满足条件的三元组,将其存入结果数组result
中。 -
在存入结果数组之前,需要判断是否重复,如果
i > 0 && nums[i] == nums[i - 1]
,说明当前找到的三元组与上一个重复,应该跳过。 -
如果
nums[left] + nums[right] == target
,则说明找到了满足条件的三元组,此时还需要继续寻找其他可能的组合,因此需要将left
和right
向中间移动,同时跳过重复的元素。 -
最后,返回结果数组。
4.总结
- 使用双指针法是一种有效的解决方法,能够在较短的时间内找到满足条件的解。
- 通过排序和双指针法,可以高效地解决这类问题,降低时间复杂度。