18. 四数之和 - 力扣

a篇博客就溜2024-06-14 20:23

1. 题目

给你一个由 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

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

2. 示例

3. 分析

做这题之前先做这道：三数之和，对应题解：三数之和 - 题解

四数之和无非就是再多套一层循环，即再增加一个固定数。之后就利用双指针寻找两数之和 == target - 第一个固定数 - 第二个固定数：

cpp 复制代码

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());

        vector<vector<int>> ret;
        int n = nums.size();

        for(int i = 0; i < n;) // 第一个固定数
        {
            for(int j = i + 1; j < n;) // 第二个固定数
            {
                // 双指针
                int left = j + 1, right = n - 1;
                long long aim = (long long)target - nums[i] - nums[j];
                while(left < right)
                {
                    int sum = nums[left] + nums[right];
                    if(sum > aim) right--;
                    else if(sum < aim) left++;
                    else
                    {
                        ret.push_back({nums[i], nums[j], nums[left++], nums[right--]});
                        while(left < right && nums[left] == nums[left-1]) left++; // 去重左指针元素
                        while(right < right && nums[right] == nums[right+1]) right--; // 去重右指针元素
                    }
                }
                j++;
                while(j < n && nums[j] == nums[j-1]) j++; // 去重第二个固定数指针元素
            }
            i++;
            while(i < n && nums[i] == nums[i-1]) i++; // 去重第一个固定数指针元素
        }
        return ret;
    }
};