三数之和
题目描述
给你一个整数数组 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[n],由此我们就把三数和问题变成了两数和问题。
那么对于两数和的问题,同时这个数组已经有序了,我们可以使用双指针的思路,定义两个指针 left 和 right,然后进行对撞遍历,去看两个指针指向的元素之和是否等于 -nums[n],如果等于,那么就把 n,left 和 right 位置的元素插入到目标数组里面;如果两个元素之和小于 -nums[n],那么就让 left 指针向前走,因为数组是有序的;如果两个元素之和大于 -nums[n],那么就让 right 指针向后走;但是这里我们还有一个问题,就是如何去重,如果我们找到了等于 -nums[n] 的两个元素,我们这两个指针还要继续进行对撞遍历,并进行去重的操作,目的是为了将这个区间内的所有符合要求的元素都找出来。
接下来我们就来讨论第三个元素,我们发现,如果我们只进行一次操作,无法找到所有满足要求的元素,所以我们需要进行循环遍历,也就是说,我们需要定义一个循环,让这个 n 走完整个数组。
根据上面的分析,我们可以写下如下代码。
代码编写
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(), nums.end());
int n = nums.size() - 1;
vector<vector<int>> result;
while (n >= 2) {
if (nums[n] < 0) {
break;
}
int left = , right = n - ;
target = -nums[n];
(left < right) {
(nums[left] + nums[right] > target) {
--right;
} (nums[left] + nums[right] < target) {
++left;
} {
result.({nums[n], nums[left], nums[right]});
++left;
(left < right && nums[left] == nums[left - ]) {
++left;
}
--right;
(left < right && nums[right] == nums[right + ]) {
--right;
}
}
}
--n;
(n >= && nums[n] == nums[n + ]) {
--n;
}
}
result;
}
};


