Leetcode 18. 四数之和

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

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

示例 1：

输入：nums = [1,0,-1,0,-2,2], target = 0
输出：[[-2,-1,1,2],[-2,0,0,2],[-1,0,0,1]]
示例 2：

输入：nums = [], target = 0
输出：[]

提示：

0 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109

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

C++

class Solution {
    /*
    和15题相同思路，排序，循环遍历，双指针。
    时间复杂度O(n^3)
    */
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> res;
        if(nums.size()<4)
            return res;
        sort(nums.begin(),nums.end());
        int len=nums.size();
        for(int i=0;i<len-3;i++){
            //选第一个数
                //避免第一个数重复
            if(i>0 && nums[i]==nums[i-1])
                continue;
                //特殊情况
            if(nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target)
                break;
            if(nums[i]+nums[len-1]+nums[len-2]+nums[len-3]<target)
                continue;

            //选第二个数
        
            for(int j=i+1;j<len-2;j++){
                //避免第二个数重复
                if(j>i+1 &&  nums[j]==nums[j-1])
                    continue;   
                if(nums[i]+nums[j]+nums[j+1]+nums[j+2]>target)
                    break;
                if(nums[i]+nums[j]+nums[len-1]+nums[len-2]<target)
                    continue;
                int l=j+1;   //双指针选择第3、4个元素
                int r=len-1;
                while(l<r){
                    int sum=nums[i]+nums[j]+nums[l]+nums[r];
                    if(sum==target){
                        res.push_back({nums[i],nums[j],nums[l],nums[r]});
                        //避免第三个数重复
                        while(l<r && nums[l]==nums[l+1])
                            l++;
                        l++;
                        while(l<r && nums[r]==nums[r-1])  //避免第四个数重复
                            r--;
                        r--;
                    }else if(sum<target){
                        l++;
                    }  
                   else{
                    r--;
                   }  
                }
            }
                      
        }
        return res;
    }
};