<18>——4Sum

18、4Sum

四数之和

给一个包含n个数的整数数组S，在S中找到所有使得和为给定整数target的四元组(a, b, c, d)。

 注意事项

四元组(a, b, c, d)中，需要满足a <= b <= c <= d

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

样例

例如，对于给定的整数数组S=[1, 0, -1, 0, -2, 2] 和 target=0. 满足要求的四元组集合为：

(-1, 0, 0, 1)

(-2, -1, 1, 2)

(-2, 0, 0, 2)

分析：在三元组的情况下增加了一组循环。

代码：

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> res;
        if(nums.size()<4)return res;
        for(int i=0;i<nums.size()-3;i++)
        {
            for(int j=i+1;j<nums.size()-2;j++)
            {
                int left=j+1,right=nums.size()-1;
                while(left<right)
                {
                    int sum=nums[i]+nums[j]+nums[left]+nums[right];
                    if(sum<target)left++;
                    else if(sum>target)right--;
                    else
                    {
                        vector<int> four;
                        four.push_back(nums[i]);
                        four.push_back(nums[j]);
                        four.push_back(nums[left]);
                        four.push_back(nums[right]);
                        res.push_back(four);
                        while(left<right&&nums[left]==four[2])left++;
                        while(left<right&&nums[right]==four[3])right--;
                    }
                }
                while(j+1<nums.size()-2&&nums[j+1]==nums[j])j++;
            }
            while(i+1<nums.size()-3&&nums[i+1]==nums[i])i++;
        }
        return res;
    }
};