力扣15 三数之和

给你一个整数数组 nums ，判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ，同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

注意：答案中不可以包含重复的三元组。

去重的逻辑有点东西。

我的想法是用哈希，但是超时了。超时版本：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        multimap<int,vector<int>>mp;
        set<vector<int>>st;
        int len=nums.size();
        multimap<int,vector<int>>::iterator it,itt;
        for(int i=0;i<len;i++){
            for(int j=i+1;j<len;j++){
                vector<int>vec;
                vec.push_back(i);
                vec.push_back(j);
                mp.insert(pair(nums[i]+nums[j],vec));
                vec.clear();
            }
        }
        for(int i=0;i<len;i++)
        {
            int s=-nums[i];
            it=mp.find(s);
            if(it!=mp.end())
            for(itt=it;itt->first==it->first;itt++)
            if(itt->second[0]!=i&&itt->second[1]!=i){
                    vector<int>ans;
                    ans.push_back(nums[itt->second[0]]);
                    ans.push_back(nums[itt->second[1]]);
                    ans.push_back(nums[i]);
                    sort(ans.begin(),ans.end());
                    st.insert(ans);
                    ans.clear();
            }
        }
        set<vector<int>>::iterator q;
        vector<vector<int>> vec;
        for(q=st.begin();q!=st.end();q++){
            vec.push_back(vector<int>(q->begin(),q->end()));
        }
        return vec;      
    }
};

双指针法（我感觉可以叫三指针法）

i循环遍历整个nums数组，l=i+1，r=len-1。判断当前的三个位置满足的条件，进行移动。

关键在于对a,b,c三个位置的减枝。

每个位置的剪枝都需要注意是这个结果用完了才能剪枝。

a位置：当第一个位置的结果集都处理完了，与他相同的数就没必要处理了，因为都是处理第一个位置的子集了。因此是

b,c位置的剪枝比较好理解。当结果集已经纳入了，如果i+1的位置和i相同对于同一个a位置就没必要再处理一次了。j-1的位置和j相同 也没必要再处理了。

我的做法仅仅只剪枝了b,c位置，没考虑a位置的剪枝，最后再用set进行去重，另外vector赋初值的方法要掌握一下。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        int len=nums.size();
        set<vector<int>>vec;
        set<vector<int>>::iterator it;
        set<int>st;
        sort(nums.begin(),nums.end());
        for(int i=0;i<len;i++){
            int l=i+1;
            int r=len-1;
            while(l<r&&l<len&&r>0)
            {
                if(nums[i]+nums[l]+nums[r]>0)
                   r--;  
                else if(nums[i]+nums[l]+nums[r]<0)
                   l++;
            else
            {
                vector<int>ans;
                ans.push_back(nums[i]);
                ans.push_back(nums[l]);
                ans.push_back(nums[r]);
                sort(ans.begin(),ans.end());
                vec.insert(ans);
                int ll=l+1;
                int rr=r-1;
                while(nums[ll]==nums[l]&&ll<=len-2)
                     ll++;
                l=ll;
                while(nums[rr]==nums[r]&&rr>=1)
                    rr--
                r=rr;
            }
        }
        }
        vector<vector<int>>ans;
        for(it=vec.begin();it!=vec.end();it++){
            vector<int>p(it->begin(),it->end());
            ans.push_back(p);
        }
        return ans;
    }
};

代码随想录的代码：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> result;
        sort(nums.begin(), nums.end());
        for (int i = 0; i < nums.size(); i++) {
            if (nums[i] > 0) {
                return result;
            }
            if (i > 0 && nums[i] == nums[i - 1]) {
                continue;
            }//当第一个位置的结果集都处理完了，与他相同的数就没必要处理了，因为都是第一个位置的子集了
            int left = i + 1;
            int right = nums.size() - 1;
            while (right > left) {
                // 去重复逻辑如果放在这里，0，0，0 的情况，可能直接导致 right<=left 了，从而漏掉了 0,0,0 这种三元组
                /*
                while (right > left && nums[right] == nums[right - 1]) right--;
                while (right > left && nums[left] == nums[left + 1]) left++;
                */
                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]});  //vector赋初值的方法学一下
                    // 去重逻辑应该放在找到一个三元组之后，对b 和 c去重
                    while (right > left && nums[right] == nums[right - 1]) right--;
                    while (right > left && nums[left] == nums[left + 1]) left++;

                    // 找到答案时，双指针同时收缩
                    right--;
                    left++;
                }
            }
        }
        return result;
    }
};

 

 

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/3sum