题目描述
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?找出所有满足条件且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例:
给定数组 nums = [-1, 0, 1, 2, -1, -4],
满足要求的三元组集合为:
[
[-1, 0, 1],
[-1, -1, 2]
]
简单分析
只能想到最暴力的解法,然后O(n^3)肯定是不行的,去参考了一下题解,发现一种优化是用排序+双指针,这就意味着我们需要给排序后的元素找凑零的两个解,而且要求不重复,那么大于0的元素可以把不考虑了,因为他们本身肯定已经被前面的组合了,只考虑小于等于0的元素,这时可以设置双指针,一个从左向右遍历,一个从右向左,左+右+当前>0,说明右边的过大,右指针向左滑动,小于0左指针向右滑动,去重的话即为相等元素直接跳过,若左右相遇,且仍未找到=0,则本次指针遍历结束,为下一元素重复开始遍历。
代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
int len = nums.size();
if(len<3) return res;
sort(nums.begin(),nums.end());
if(nums[0]>0||nums[len-1]<0) return res;
for(int i=0;i<len-1;i++){
if(nums[i]>0)
break;
int target = 0 - nums[i];
int left = i+1;
int right = len-1;
while(left < right){
if(nums[right] < 0)
break;
if(nums[left] + nums[right] < target){
int v= nums[left];
while(left != right && nums[left] == v)
left++;
}
else if(nums[left] + nums[right] > target){
int v= nums[right];
while(left != right && nums[right] == v)
right --;
}else{
vector<int> tmp_res{nums[i],nums[left],nums[right]};
res.push_back(tmp_res);
int v= nums[left];
while(left != right && nums[left] == v)
left++;
v= nums[right];
while(left != right && nums[right] == v)
right --;
}
}
while(i<len-1&&nums[i] == nums[i+1]) i++;
}
return res;
}
};