leetcode18.四数之和

给定一个包含 n 个整数的数组 nums 和一个目标值 target，判断 nums 中是否存在四个元素 a，b，c 和 d ，使得 a + b + c + d 的值与 target 相等？找出所有满足条件且不重复的四元组。

注意：

答案中不可以包含重复的四元组。

示例：

给定数组 nums = [1, 0, -1, 0, -2, 2]，和 target = 0。

满足要求的四元组集合为：
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/4sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

完整代码

这道题的难点在于如何去重，并且保证不超时，这里采用集合的方法去重。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        set<vector<int>> res;
        vector<vector<int>> result;
        if(nums.size()<4)
            return result;
        vector<int> c;
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size()-3;++i){
            //if(i!=0&&nums[i]==nums[i-1])
              //  continue;
            for(int j=i+1;j<nums.size()-2;++j){
                //if(j!=nums.size()-3&&nums[j]==nums[j-1])
                    //continue;
                int s=j+1;
                int e=nums.size()-1;
                while(s<e){
                    int cur=nums[i]+nums[j]+nums[s]+nums[e];
                    if(cur==target){
                        c.clear();
                        c={nums[i],nums[j],nums[s],nums[e]};
                        res.insert(c);
                        ++s;
                        // while(s<e&&nums[s]==nums[s-1])
                        //     ++s;
                    }
                    else if(cur>target){
                        --e;
                        // while(s<e&&nums[e]==nums[e+1])
                        //     --e;
                    }
                    else{
                        ++s;
                        // while(s<e&&nums[s]==nums[s-1])
                        //     ++s;
                    }
                }
            }
        }
        
        for(auto c:res)
            result.push_back(c);
        return result;
    }
};

如何利用跳过重复数字的方法去重呢？？？
其实是可以的，参见代码

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> res;
        if(nums.size()==0)
            return res;
        vector<int> c;
        sort(nums.begin(),nums.end());
        for(int i=0;i<nums.size();++i){
            if(i!=0&&nums[i]==nums[i-1])//去重，这里不要忘记i!=0，不然的话会出现数组越界
                continue;
            for(int j=i+1;j<nums.size();++j){
                if(j>i+1&&nums[j]==nums[j-1])//去重，这里不要忘记j>i+1,不然像[0,0,0,0]会通通跳过
                    continue;
                int s=j+1;
                int e=nums.size()-1;
                while(s<e){
                    int cur=nums[i]+nums[j]+nums[s]+nums[e];
                    if(cur==target){
                        c.clear();
                        c={nums[i],nums[j],nums[s],nums[e]};
                        res.push_back(c);
                        ++s;
                        while(s<e&&nums[s]==nums[s-1])//去重
                            ++s;
                    }
                    else if(cur>target){
                        --e;
                        while(s<e&&nums[e]==nums[e+1])//去重
                            --e;
                    }
                    else{
                        ++s;
                        while(s<e&&nums[s]==nums[s-1])
                            ++s;
                    }
                }
            }
        }
        return res;
    }
};