给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
暴力破解
问题:如何解决相同答案问题?—未解决
///错误答案
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int> >sum;
int len = nums.size();
if(len<3)
return sum;
for(int i=0;i<len;i++){
for(int j =i+1;j<len;j++){
for(int k=j+1;k<len;k++){
if(nums[i]+nums[j]+nums[k]==0)
{
sum.push_back(vector<int>{nums[i],nums[j],nums[k]});
}
}
}
}
return sum;
}
};
排序+双指针
只有排序后,指针的移动才有意义
///第一次代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int> >ans;
int sum;
int len = nums.size();
sort(nums.begin(), nums.end());
if(len<3)
return ans;
for (int t = 0; t < len; t++){
int j=len-1;
int i=t+1;
sum=nums[t]+nums[i]+nums[j];
if(i>=j)
continue;
if(sum>0){
j--;
sum=nums[t]+nums[i]+nums[j];
}
else if(sum<0){
i++;
sum=nums[t]+nums[i]+nums[j];
}
else
ans.push_back({ nums[t], nums[i], nums[j] });
}
return ans;
}
};
问题:当得到一个解时,就会进行下一次循环,而不是寻找是否还有答案;
观看答案后
要点1:由于会有一样的数字,所以一样的数字只使用一次即可,这样子就不会出现一样答案的情况
要点2:要使用while进行判断,这样子当有一个答案时,但还符合while条件时,就会继续查找
正确答案
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int> > ans;
int len=nums.size();
int i,j;
if(len<3)
return ans;
sort(nums.begin(),nums.end());
for(int t=0;t<len;t++)
{
i=t+1;
j=len-1;
if(t>0 && nums[t]==nums[t-1])
continue;
while(i<j)
{
if (nums[i] + nums[j] > -nums[t])
j--;
else if (nums[i] + nums[j] < -nums[t])
i++;
else {
ans.push_back(vector<int>{nums[t], nums[i], nums[j]});
i++;
j--;
while (i < j && nums[j] == nums[j + 1]) { j--; }
while (i < j && nums[i] == nums[i - 1]) { i++; }
}
}
}
return ans;
}
};
再次尝试暴力解法
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int> >sum;
int len = nums.size();
sort(nums.begin(),nums.end());
if(len<3)
return sum;
for(int i=0;i<len;i++){
if(i>0 && nums[i]==nums[i-1])
continue;
for(int j =i+1;j<len;j++){
if(j>i+1 && nums[j]==nums[j-1])
continue;
for(int k=j+1;k<len;k++){
if(k>j+1 && nums[k]==nums[k-1])
continue;
if(nums[i]+nums[j]+nums[k]==0)
sum.push_back(vector<int>{nums[i],nums[j],nums[k]});
}
}
}
return sum;
}
};
但是改进后,会发生运行超时。当将[1,2,-2,-1],[0,0,0],[0,0,0,0],[-1,0,1,2,-1,-4]进行输入时答案和预期结果一致