LeetCode: 18. 四数之和

最新推荐文章于 2024-07-09 14:16:21 发布

selenium1

最新推荐文章于 2024-07-09 14:16:21 发布

阅读量204

点赞数

分类专栏：力扣文章标签： leetcode 算法 c++

本文链接：https://blog.youkuaiyun.com/selenium1/article/details/123966651

版权

力扣专栏收录该内容

14 篇文章

订阅专栏

本文介绍了LeetCode中的18题——四数之和问题的解题方法。通过固定第一个数，然后使用双指针法解决剩余三个数的和等于目标值的问题。代码实现中使用了排序和迭代，确保了结果的不重复性。示例展示了在不同输入情况下，如何找到满足条件的四元组。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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

示例 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]]

提示：

1 <= nums.length <= 200
-109 <= nums[i] <= 109
-109 <= target <= 109

解题思路：四数之和的方法也可以参照三数之和来做。固定第一个数之后，后面的就参照三数之和的方法求解。（注：三数之和的方法LeetCode : 15. 三数之和_selenium1的博客-优快云博客可参照这个）。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        int len = nums.size();
        if(len < 4) return {};
        
        vector<vector<int>> res;
        sort(nums.begin(),nums.end());
        for(int a = 0; a < len - 3; a++)
        {
            if(a > 0 && nums[a] == nums[a-1]) continue;
            for(int b = a+1; b < len - 2; b++)
            {
                if(b > a+1 && nums[b] == nums[b-1]) continue;
                int c = b + 1;
                int d = len - 1;
                while(c < d)
                {
                    if(nums[a] + nums[b] == target - (nums[c] + nums[d]))
                    {
                        res.push_back({nums[a] , nums[b] , nums[c] , nums[d]});
                        while(c < d && nums[c] == nums[c + 1]) c++;
                        while(c < d && nums[d-1] == nums[d]) d--;
                        c++;
                        d--;
                    }
                    else if(nums[a] + nums[b] > target - (  nums[c] + nums[d]))
                        d--;
                    else
                        c++;
                }
            }
        }
        return res;
    }
};