LeetCode: 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

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

示例 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的博客-CSDN博客 可参照这个）。

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;
    }
};

最后

以上就是繁荣未来为你收集整理的LeetCode: 18. 四数之和的全部内容，希望文章能够帮你解决LeetCode: 18. 四数之和所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。