15. 三数之和(回溯法超时，使用双指针)

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;
    }
};