需要求三个数和为0,分解为求2个数能否何为一个固定数。
先排序,两个数能否求某个解就是简单的了。
需要注意的就是重复数字应该怎么样不都加入到结果中。
class Solution {
public:
void findTwoNums(vector<int> &num,int left,int neg_num,vector<vector<int> > &result)
{
if(left==num.size()-1) return;
else if(neg_num+num[left]+num[left+1]>0) return;
else if(neg_num+num[num.size()-1]+num[num.size()-2]<0) return;
int r=num.size()-1;
while(left<r)
{
int cur_sum=num[left]+num[r]+neg_num;
if(cur_sum==0)
{
vector<int> t;
t.push_back(neg_num);
t.push_back(num[left]);
t.push_back(num[r]);
result.push_back(t);
int rr=r;
r--;
while(num[rr]==num[r]) r--;
left++;
}
else if(cur_sum>0)
{
r--;
}
else
{
left++;
}
}
}
vector<vector<int> > threeSum(vector<int> &num) {
// Start typing your C/C++ solution below
// DO NOT write int main() function
vector<vector<int>> result;
if(num.size()<=2) return result;
sort(num.begin(),num.end());
for(int i=0;i<num.size()-2;i++)
{
if(i>0 && num[i]==num[i-1]) continue;
if(num[i]<=0)
findTwoNums(num,i+1,num[i],result);
}
return result;
}
};