问题描述
-1,2,-1 又是如何被包含在下一对中的? 问题是: 给定一个整数数组 nums,返回所有的三元组 [nums[i],nums[j],nums[k]] 使得 i != j,i != k,and j != k,and nums[i] + nums[j] + nums[k] == 0.
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<int>r1;
vector<vector<int>>resvec;
int s=nums.size();
int i=0,j,k=s-1;
for(;k-i>=2;k--)
{
for(j=i+1;j<k;j++)
{
if(nums[i]+nums[j]+nums[k]==0)
{
r1.push_back(nums[i]);
r1.push_back(nums[j]);
r1.push_back(nums[k]);
resvec.push_back(r1);
}
}
}
if(s>=3)
return resvec;
else
return {};
}
};
我期待这个输出:
[[-1,-1,2],[-1,1]]
为什么它给出这样的输出:
[[-1,-1],1]]
解决方法
您在添加新元素之前忘记清除 r1
。
if(nums[i]+nums[j]+nums[k]==0)
{
r1.clear(); // clear the vector (erase the extra elements)
r1.push_back(nums[i]);
r1.push_back(nums[j]);
r1.push_back(nums[k]);
resvec.push_back(r1);
}
您应该在内部 r1
中声明 if
,而不是清除:
if(nums[i]+nums[j]+nums[k]==0)
{
vector<int>r1; // declare here,not top of the function
r1.push_back(nums[i]);
r1.push_back(nums[j]);
r1.push_back(nums[k]);
resvec.push_back(r1);
}
C++11 或更高版本中的另一种方法是构造向量以直接添加而无需额外变量:
if(nums[i]+nums[j]+nums[k]==0)
{
resvec.push_back(vector<int>{nums[i],nums[j],nums[k]});
}