力扣--18.四数之和

题目描述

给定一个包含 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]
]

代码示例

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        // vector<vector<int>> res;
        // sort(nums.begin(), nums.end());
        // if(nums.empty()) return {};

        // for(int z=0; z<nums.size()-4; z++){
        //     //这里和之前的三数之和不同，是因为z-1已经计算过一次，避免z再重复一次，并且不需要去重，因为有可能相等
        //     if(z > 0 && nums[z] == nums[z - 1]) continue;
        //     int newtarget = target - nums[z];

        //     for(int k=z+1; k<nums.size()-3; k++){

        //         if(k > z+1 && nums[k] == nums[k - 1]) continue;
        //         int newtarget2 = newtarget - nums[k];
        //         int i = k + 1, j = nums.size()-1;
        //         while(i < j){
        //             if(nums[i] + nums[j] == newtarget2){
        //                 res.push_back({nums[z], nums[k], nums[i], nums[j]});
        //                 while(i<j && nums[i+1]==nums[i]) ++i;
        //                 while(i<j && nums[j-1]==nums[j]) --j;
        //                 ++i;
        //                 --j;
        //             }
        //             else if(nums[i] + nums[j] < newtarget2) ++i;
        //             else --j;
        //         }
        //     }
        // }
        // return res;
        sort(nums.begin(),nums.end());
        vector<vector<int>> result;
        int size=nums.size();
        for(int a=0;a<size-3;++a)
        {
            if(a>0&&nums[a]==nums[a-1])continue;
            for(int b=a+1;b<size-2;++b)//以下代码与三数之和一样的
            {
                if(b>a+1&&nums[b]==nums[b-1])continue;
                int i=b+1,j=size-1;
                while(i<j)
                {
                    int sum=nums[a]+nums[b]+nums[i]+nums[j];
                    if(sum<target)
                        while(i<j&&nums[i]==nums[++i]);
                    else if(sum>target)
                        while(i<j&&nums[j]==nums[--j]);
                    else{
                        result.push_back(vector<int>{nums[a],nums[b],nums[i],nums[j]});
                        while(i<j&&nums[i]==nums[++i]);
                        while(i<j&&nums[j]==nums[--j]);
                    }
                }
            }
        }
        return result;
    }
};