三数之和
给你一个包含 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]]
思路
先排序,再用双指针,首先遍历数组,双指针指向当前元素后的首尾位置,判断双指针指向的值之和是否==target - 当前的最小值
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> result;
sort(nums.begin(), nums.end());
int length = nums.size();
for(int i = 0; i < length - 2 && nums[i] <= 0; ++i){
if(i > 0 && nums[i] == nums[i - 1]) //去重
continue;
int left = i + 1;
int right = length -1;
while(left < right){ //双指针求第二和第三个数
int cur = nums[left] + nums[right];
if(cur == -nums[i]){ //找到一个符合的组合,则双指针去重再向中间移动
result.push_back(vector<int>{nums[i], nums[left], nums[right]});
while(left < right && nums[right] == nums[right - 1]) --right;
while(left < right && nums[left] == nums[left + 1]) ++left;
--right;
++left;
}
else if(cur > -nums[i])
--right;
else
++left;
}
}
return result;
}