上来先考虑暴力算法,如下。但是元素不相等的意思是任意子答案的结果都不能重复,顺序不同也是不可以的。比如[-1,0,1]和[-1,1,0]也是不可以的。按照代码来看,相当于一个子答案保留了三份。由此可见,本题真正的难点在于去重复。
文章目录
#include
#include
#include
using namespace std;
class Solution {
public:
vector<vector> threeSum(vector& nums) {
//bf方法:找出所有两个元素的组合,原容器中剔除这两个元素,
//以0-a-b为target搜索剩余元素。最后还要去重。
//取两个元素的组合
int numSize=nums.size();
vector<vector>total_result;
vectorone_result;
for(int i=0;i<numSize-1;++i)
{
for(int j=i+1;j<numSize;++j)
{
//nums[i] nums[j]就是所有的二元元素
//循环遍历找另外一个元素
int twoSum=nums[i]+nums[j];
for(int k=0;k<numSize;++k)
{
if(nums[k] -twoSum&&k!=i&&k!=j)
{
one_result.push_back(nums[i]);
one_result.push_back(nums[j]);
one_result.push_back(nums[k]);
vector<vector>::iterator it=find(total_result.begin(),total_result.end(),one_result);
if(ittotal_result.end())
{
total_result.push_back(one_result);
one_result.clear();
}
}
}
}
}
return total_result;
}
};
题解给出了很好的解决重复的方法,固定一个元素,另外两维护两个指针,三个数相加。写程序之前,一定要先考虑特判。对特定元素的去重,一定要使用第一个,把后面重复的去掉,原因是left是从特定元素之后的一个元素开始的。
文章目录
#include
#include
#include
using namespace std;
class Solution {
public:
vector<vector> threeSum(vector& nums) {
int numSize=nums.size();
vectorone_ans;
vector<vector>total_ans;
sort(nums.begin(),nums.end());
//特判
if(numSize<3 || nums[0]>0) return total_ans;
for(int i=0;i<numSize-1;++i)
{
//排序数组 去重
if(i>0 && nums[i]==nums[i-1]) continue;
//特判
if(nums[i]>0) break;
int left(i+1),right(numSize-1);
while(left<right)
{
if(nums[i]+nums[left]+nums[right]==0)
{
one_ans.push_back(nums[i]);
one_ans.push_back(nums[left]);
one_ans.push_back(nums[right]);
total_ans.push_back(one_ans);
one_ans.clear();
//一个nums[i]可能对应着好几组left和right
//所以要继续变化left/right
do
{
++left;
}while(left<right && nums[left-1]==nums[left]);
do
{
--right;
} while (left<right && nums[right]==nums[right+1]);
}
else if(nums[i]+nums[left]+nums[right]<0)
++left;
else if(nums[i]+nums[left]+nums[right]>0)
--right;
}
}
return total_ans;
}
};