给定一个包含 n 个整数的数组 nums
,判断 nums
中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
思路1:仿照两数之和,不需要排序,但是会返回重复复的三元组,因此要去重,用标准库unique函数出重,仍通不过最后一个很多零的测试用例,但是还是把代码贴一下;
思路2:先排序,然后用滑动窗口法,找匹配的三元组,在数组值相同时,窗口左右游标滑动,来达到去重目的,排序复杂度nlogn,遍历两次数组,复杂度n的平方,总复杂度n的平方。
/////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
class Solution {
public:
vector<vector<int> > threeSum(vector<int> &num) {
vector<vector<int> > res;
std::sort(num.begin(), num.end());
for (int i = 0; i < num.size(); i++) {
int target = -num[i];
int front = i + 1;
int back = num.size() - 1;
while (front < back) {
int sum = num[front] + num[back];
if (sum < target)
front++;
else if (sum > target)
back--;
else {
vector<int> triplet(3, 0);
triplet[0] = num[i];
triplet[1] = num[front];
triplet[2] = num[back];
res.push_back(triplet);
while (front < back && num[front] == triplet[1]) front++;
while (front < back && num[back] == triplet[2]) back--;
}
}
while (i + 1 < num.size() && num[i + 1] == num[i])
i++;
}
return res;
}
};
///////////////////////////
原文地址:https://www.cnblogs.com/zzas0/p/10525483.html