15. 三数之和
题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
Code
class Solution {
/*
//此题使用回溯会超时
//对回溯加深理解:使用vector<bool> &used的目的是对于元素的内部排序有要求
// 使用去重时,记得i>start
private:
vector<vector<int>> result;
vector<int> temp;
void backTracking(vector<int> &nums,int start){
if(temp.size()==3){
int sum=temp[0]+temp[1]+temp[2];
if(sum==0) {
result.push_back(temp);
}
}
for(int i=start;i<nums.size();i++){
if(i>start&&nums[i]==nums[i-1]) continue;
//if(used[i]==true) continue;
temp.push_back(nums[i]);
//used[i]=true;
backTracking(nums,i+1);
temp.pop_back();
//used[i]=false;
}
}
public:
vector<vector<int>> threeSum(vector<int>& nums) {
if(nums.size()<3) return {};
result.clear();
temp.clear();
sort(nums.begin(),nums.end());
vector<bool> used(nums.size(),false);
backTracking(nums,0);
return result;
}
*/
public:
vector<vector<int>> threeSum(vector<int>& nums) {
if(nums.size()<3) return {};//小于三个数,则直接返回空
vector<vector<int>> result;
sort(nums.begin(),nums.end());//从小到大排序
for(int i=0;i<nums.size();i++){
if(nums[i]>0) break;//当第一个数字大于零的时候,由于时从小到大的排序,则不会有符合条件的vector直接break
if(i>0&&nums[i]==nums[i-1]) continue;//去重
int left=i+1;//左指针
int right=nums.size()-1;//右指针
while(left<right){
if(nums[i]+nums[left]+nums[right]>0) right--;
else if(nums[i]+nums[left]+nums[right]<0) left++;
else{
result.push_back(vector<int>{nums[i],nums[left],nums[right]});
left++;
right--;
while(left<right&&nums[left]==nums[left-1]) left++;//去重
while(left<right&&nums[right]==nums[right+1]) right--;//去重
}
}
}
return result;
}
};