Given an array S of n integers, are there elements a, b, c in S such that a b c = 0? Find all unique triplets in the array which gives the sum of zero.
Note: The solution set must not contain duplicate triplets.
代码语言:javascript复制For example, given array S = [-1, 0, 1, 2, -1, -4],
A solution set is:
[
[-1, 0, 1],
[-1, -1, 2]
]
题目不难,意外发现一个容易被忽视的C 小Tips。
题意为找出和为0的三元组,按升序排列,不重复。
首先肯定是排序,
然后我习惯性想到枚举两个数,最后一个用map或者hash来表示是否存在,
然而这样不是和之前某两个数求和的问题一样了吗?
Leetcode显然不会这么无聊,于是我想了很久,无意中发现两点法可行,
复杂度和第一种用hash的情况一样,都为n方,但是实现起来要巧妙很多,
枚举第一个数,以第一个数后边的一个数和末尾的数为两点,往中间缩,
如果三者的和大于零,说明最大值需要小一些,因为最小值已经无法更小了,就将末尾前移,
反之,则将头部后移。
等于零的时候记录一下,继续找。
写好后调试了一会儿,发现了无法解释的BUG,
代码语言:javascript复制nums.size()-2 这一句单独拿出来,值非常奇怪,需要借助int变量才能变回去!
代码语言:javascript复制我搞ACM的时候一直是这么写的啊!
后来经过观察,我发现vector size()的返回值为一个无符号整形,
因为ACM题的数据范围一般不会让它出现负数的情况,所以我一直都没有发现,
而这里-2显然在输入为空的时候,会让无符号整形溢出,好的,问题解决
又学到了一个小细节。
详情请看代码
代码语言:javascript复制class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
sort(nums.begin(),nums.end());
vector<vector<int>> result;
int len=(nums.size()-2);
for(int i=0;i<len;i )
{
//cout<<i<<" "<<nums.size()-2<<endl; 就是这里RE发现问题
if(nums[i]>0)continue;
if(i!=0 && nums[i]==nums[i-1])continue;
int l=i 1,r=nums.size()-1;
while(l<r)
{
int com=nums[i] nums[l] nums[r];
if(com>0)
r--;
else if(com<0)
l ;
else
{
result.push_back({nums[i], nums[l], nums[r]});
while(l<r && nums[l]==nums[l 1]) l ;
while(l<r && nums[r]==nums[r-1]) r--;
r--;
l ;
}
}
}
return result;
}
};