【leetcode】15. 3 Sum 双指针 压缩搜索空间

　　Given an integer array nums, return all the triplets [nums[i], nums[j], nums[k]] such that i != j, i != k, and j != k, and nums[i] + nums[j] + nums[k] == 0. Notice that the solution set must not contain duplicate triplets.

　　

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        
        vector<vector<int>> res={};
        
        if(nums.size()<3) return res;
        // 暴力发 o(n3) cn/3 如何抽取这三个元素
        // 如何保证插入的元素 不重复 多用set
        // 双指针的改进版本 这个题目 有点意思
        sort(nums.begin(),nums.end());// 压缩搜索空间
        set<vector<int>> dp;
        
        int n=nums.size();
        for(int i=0;i<n;++i){
            int left=i+1;
            int right=n-1;
            while(left<right){
                if((nums[i]+nums[left]+nums[right])==0){
                    dp.insert({nums[i],nums[left],nums[right]});
                    left++;
                    right--;
                }
                else if((nums[i]+nums[left]+nums[right])<0){
                    left++;
                }
                else if((nums[i]+nums[left]+nums[right])>0){
                    right--;
                }
                
            }
        }
        for(auto dd:dp){
            res.push_back(dd);
        }
        return res;
        
        
    }
};