这道题两个难点。第一个是不能使用三层循环,因为会时间溢出,第二个也是主要的问题如何去重。一开始想要直接排序后去重,但是这样的话会缺失解,比如{-1,-1,2}这个解就得不到,所以不能直接去重。后来想到了可以判断nums[i]与nums[i-1]是否相同,如果相同可以进行i++,这样就可以去重了,但这只是死一次去重,要用不同的三个数,所以要去进行三次去重。【这个在后面解释】。三层循环不能用就化简为2层循环,使用类似左右指针的形式在确定第一个数的情况下,遍历剩下的数组元素。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
int a;
vector<vector<int>> result;
sort(nums.begin(), nums.end());
for(int i=0;i<nums.size();i++)
{
if(i>0&&nums[i]==nums[i-1]) continue;//第一次去重
if((a=nums[i]) > 0) break;
int l=i+1,r=nums.size()-1;//设置左右指针遍历
while(l<r)
{
if (nums[l] + nums[r] + a < 0) ++l;
else if (nums[l] + nums[r] + a > 0) --r;
else{
result.push_back({a, nums[l], nums[r]});
++l,--r;
while (l < r && nums[l] == nums[l - 1]) ++l;//第二次去重
while (l < r && nums[r] == nums[r + 1]) --r;//第三次去重
}
}
}
return result;
}
};
nums【l】!=nums[l-1]右面的可以多次用,但是如果nums【l】==nums[l-1],还在继续使用的话,就会产生重复。r同理。