和3Sum很类似,外面多了一层循环。考虑数字重复的情况。
首先,对所有数据进行排序,然后基本代码可以使用3Sum的。
class Solution {
public:
void findTwoNums(vector<int> &num,int left,int neg_num,vector<vector<int> > &result,
int &first_num,int &second_num)
{
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(first_num);
t.push_back(second_num);
t.push_back(num[left]);
t.push_back(num[r]);
result.push_back(t);
int rr=r;
r--;
while(r>left && num[rr]==num[r]) r--;
left++;
}
else if(cur_sum>0)
{
r--;
}
else
{
left++;
}
}
}
vector<vector<int> > fourSum(vector<int> &num, int target) {
// Start typing your C/C++ solution below
// DO NOT write int main() function
vector<vector<int>> result;
if(num.size()<=3) return result;
sort(num.begin(),num.end());
for(int i=0;i<num.size()-3;i++)
{
for(int j=i+1;j<num.size()-2;j++)
{
if(i>0 && num[i]==num[i-1]) continue;
if(j-1!=i && num[j-1]==num[j]) continue;
int two_sum=num[i]+num[j];
//if(two_sum-target<=0)
findTwoNums(num,j+1,two_sum-target,result,num[i],num[j]);
}
}
return result;
}
};