先枚举前两个数,再看第三个数在不在,在这里主要需处理好重复的问题,所以先排好序,若第一个数字相同,跳过,若第二个数字想同,也跳过
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int n=nums.size();
int i,j,k,length;
int ans;
sort(nums.begin(),nums.end());
unordered_map<int,int> mp;
vector<vector<int>> res;
for(i=0;i<n;i++)
mp[nums[i]]=i;
for(i=0;i<=n-3;i++)
{
if(i>0&&nums[i]==nums[i-1])
continue;
for(j=i+1;j<n;j++)
{
if(j>i+1&&nums[j]==nums[j-1])
continue;
int d=nums[i]+nums[j];
if(d>0)
break;
if(mp.find(-d)!=mp.end())
{
k=mp[-d];
if(k<=j)
break;
vector<int> vt;
vt.push_back(nums[i]);
vt.push_back(nums[j]);
vt.push_back(nums[k]);
res.push_back(vt);
}
}
}
return res;
}
};
另一种方法,不需要空间开销,二分查找,枚举第一个数,第一个数从左端开始找,第二个数从右端开始找
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int n=nums.size();
int i,j,k;
sort(nums.begin(),nums.end());
vector<vector<int>> res;
for(i=0;i<=n-3;i++)
{
if(i>0&&nums[i]==nums[i-1])
continue;
for(j=i+1,k=n-1;j<k;)
{
if(j>i+1&&nums[j]==nums[j-1])
{
j++;
continue;
}
if(k<n-1&&nums[k]==nums[k+1])
{
k--;
continue;
}
int d=nums[i]+nums[j]+nums[k];
if(d==0)
{
vector<int> vt;
vt.push_back(nums[i]);
vt.push_back(nums[j]);
vt.push_back(nums[k]);
res.push_back(vt);
j++;
k--;
}
else if(d>0)
k--;
else if(d<0)
j++;
}
}
return res;
}
};