【算法day17】四数之和—给你一个由 n 个整数组成的数组 nums ，和一个目标值 target 。请你找出并返回满足下述全部条件且不重复的四元组 [nums[a], nums[b], nums[

原创于 2025-03-21 23:20:06 发布 · 349 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构

算法专栏收录该内容

90 篇文章

订阅专栏

18. 四数之和

https://leetcode.cn/problems/4sum/description/

给你一个由 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
你可以按任意顺序返回答案。

示例 1：

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

输入：nums = [2,2,2,2,2], target = 8
输出：[[2,2,2,2]]

在这里插入图片描述

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(), nums.end());
        vector<vector<int>> ans;
        int nums_len = nums.size();
        // 尽可能减枝
        for (int i = 0; i < nums_len - 3; i++) {
            if (i > 0 && nums[i] == nums[i - 1])
                continue;
            if ((long)nums[i] + nums[i + 1] + nums[i + 2] + nums[i + 3] > target)
                break;
            for (int j = i + 1; j < nums_len - 2; j++) {
                if (j > i + 1 && nums[j] == nums[j - 1])
                    continue;
                if ((long)nums[i] + nums[j] + nums[j + 1] + nums[j + 2] > target)
                    break;
                int d = nums_len - 1;
                for (int k = j + 1; k < nums_len - 1; k++) {
                    if (k > j + 1 && nums[k] == nums[k - 1])
                        continue;
                    if ((long)nums[i] + nums[j] + nums[k] + nums[k + 1] > target)
                        break;
                    long aim_num = (long)target - nums[i] - nums[j] - nums[k];
                    while (d > k && nums[d] > aim_num) {
                        d--;
                    }
                    if (d == k)
                        break;
                    if (nums[d] == aim_num) {
                        ans.push_back({nums[i], nums[j], nums[k], nums[d]});
                    }
                }
            }
        }
        return ans;
    }
};