给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
思路:
1、排除三个数都为0的情况,也就是[0,0,0];请他情况必定是有一个负数的。所以我们可以首先就排个序。
2、遍历排序后的负数所在的范围
3、拿到第一个负数后,我们只需要再拿到后面两个数,与之相加=0即可。后面两个数我们用两个指针来表示,j和k,一个是从左边往右边走,一个是从最右边往左边走。先排好序,左边一个left指针,右边一个right指针,如果left+right > result,那么right往左移,如果left+right < result,那么说明相加得数不够,就需要left往右移。找到相加之和相等的就行
4、两个while滤掉重复的值
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<int> numSet(3);
vector< vector<int> > r;
// 1.排序
sort(nums.begin(), nums.end());
int sum;
int len = nums.size();
// 2.拿出第一个数,转化为两数和问题。注意外层循环到倒数第三个数即可
for(int i = 0; i < len-2; i++) {
sum = 0 - nums[i];
numSet[0] = nums[i];
// 3.两数和问题
for(int j = i+1, k = len-1; j < k;) {
if(nums[j] + nums[k] == sum) {
numSet[1] = nums[j++];
numSet[2] = nums[k--];
r.push_back(numSet);
// 根据题目要求,跳过重复元素
while(j < k && nums[j] == nums[j-1])
j++;
while(j < k && nums[k] == nums[k+1])
k--;
}
else if(nums[j] + nums[k] < sum)
j++;
else
k--;
}
while(i < len-2 && nums[i+1] == nums[i])
i++;
}
return r;
}
};```