题目:
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:给定数组 nums = [-1, 0, 1, 2, -1, -4],满足要求的三元组集合为:[ [-1, 0, 1], [-1, -1, 2]]
思路:
开始单纯用for循环,用时过长,参考了别人的解题方式
class Solution {
public: vector<vector<int>> threeSum(vector<int>& nums)
{ vector<vector<int>>res;
int i,j,k;
if(nums.size()<3)
return res;
sort(nums.begin(),nums.end()); for(i=0;i<nums.size()-2;i++)
{
if(nums[i]>0)
break;
if(i>0&&nums[i]==nums[i-1])
continue;
j=i+1;
k=nums.size()-1;
do{
int s=nums[i]+nums[j]+nums[k]; if(s>0)
--k;
else if(s<0)
++j;
else {res.push_back({nums[i],nums[j],nums[k]}); while(k>j&&nums[k]==nums[--k]); while(k>j&&nums[j]==nums[++j]);} }while(k>j);
}
return res;
}};