18. 四数之和 ●●

18. 四数之和 ●●

描述

给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[c], nums[d]] （若两个四元组元素一一对应，则认为两个四元组重复）：

• 0 <= a, b, c, d < n
• a、b、c 和 d 互不相同
• nums[a] + nums[b] + nums[c] + nums[d] == target

你可以按 任意顺序 返回答案 。

示例

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

题解

1. 排序 + 双指针

与 15. 三数之和 解法类似，多套一层 for 循环，但是要注意此时的剪枝操作，不要判断 nums[i] > target 就返回了，因为 target 是任意值。

• 时间复杂度：$O(n^3)$，其中 n 是数组的长度。排序的时间复杂度是 $O(nlogn)$，枚举四元组的时间复杂度是 $O(n^3)$

• 空间复杂度：O(logn)，其中 n 是数组的长度。空间复杂度主要取决于排序额外使用的空间。此外排序修改了输入数组 nums，实际情况中不一定允许，因此也可以看成使用了一个额外的数组存储了数组 nums 的副本并排序，空间复杂度为 O(n)。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> ans;
        int n = nums.size();
        if(n < 4) return ans;
        sort(nums.begin(), nums.end());
        for(int i = 0; i < n - 3; ++i){
            if(nums[i] > target && nums[i] > 0) break;                  //  剪枝
            if(i > 0 && nums[i] == nums[i-1]) continue;                 // a去重
            for(int j = i + 1; j < n - 2; ++j){
                if(nums[i] + nums[j] > target && nums[j] > 0) break;    //  剪枝
                if(j > i + 1 && nums[j] == nums[j-1]) continue;         // b去重
                int l = j + 1;
                int r = n - 1;
                while(l < r){                                           // 查找c，d
                    if(nums[i] + nums[j] > target - nums[l] - nums[r]){ // 防止溢出
                        --r;
                    }else if(nums[i] + nums[j] < target - nums[l] - nums[r]){
                        ++l;
                    }else{
                        ans.emplace_back(vector<int>{nums[i], nums[j], nums[l], nums[r]});
                        while(l < r && nums[l] == nums[l + 1]){
                            ++l;
                        }
                        ++l;
                    }
                }
            }
        }
        return ans;
    }
};