题目大意:在数组中找到所有三个数之和为0的组合。
首先对数组排序,然后用两个循环求数组中两个不同数的和sum2,然后再在数组的剩余部分找-sum2的位置即可。时间复杂度n2
要小心的就是出现连续相同的数的情况,比如联系三个或以上的数相同这是就需要想办法避免重复计算了。
下面是AC代码
class Solution {
public:
int locate(vector<int> &num,int low,int high,int target)
{
if(low>high)return -1;
if(target<num[low] || target > num[high])return -1;
int mid=(low+high)/2;
if(num[mid]==target)return mid;
if(num[mid]<target)return locate(num,mid+1,high,target);
return locate(num,low,mid-1,target);
}
vector<vector<int> > threeSum(vector<int> &num) {
sort(num.begin(),num.end());
int len=num.size();
vector<vector<int>> result;
for(int i=0;i<len-1;i++)
{
if(i==0 || (i>0 && num[i]!=num[i-1]))
{
for(int j=i+1;j<len;j++)
{
if(j==i+1 || (j>i+1 && num[j]!=num[j-1]))
{
int sum2=num[i]+num[j];
int third=locate(num,j+1,len-1,-sum2);
if(third>0)
{
vector<int> tmp;
tmp.push_back(num[i]);
tmp.push_back(num[j]);
tmp.push_back(num[third]);
result.push_back(tmp);
};
}
}
}
};
return result;
}
};