class Solution {
public:
vector<vector<int> > threeSum(vector<int> &num) {
vector<vector<int>> ret;
//如果数组元素小于3个,则直接返回
if(num.size() < 3)
return ret;
//先给整个数组排序
sort(num.begin(), num.end());
auto it1 = num.begin();//指向第一个元素
while(it1 != num.end() - 2)
{
auto it2 = it1 + 1;//第二个元素,向后遍历
auto it3 = num.end() - 1;//第三个元素,向前遍历
while(it2 < it3)
{
if(*it1+*it2+*it3 > 0)
it3--;
else if(*it1+*it2+*it3 < 0)
it2++;
else
{
//符合条件则插入
vector<int> temp;
temp.push_back(*it1);
temp.push_back(*it2);
temp.push_back(*it3);
ret.push_back(temp);
//去重
while(it2+1<it3 && *(it2+1)==*it2)
it2++;
while(it3-1>it2 && *(it3-1)==*it3)
it3--;
it2++, it3--;
}
}
//去重
while(it1+1!=num.end()-2 && *(it1+1)==*it1)
it1++;
it1++;
}
return ret;
}
};
数组中相加和为0的三元组
最新推荐文章于 2022-08-06 02:21:23 发布