Leetcode每日一题:18. 四数之和(2023.7.15 C++)

目录

18. 四数之和

题目描述:

实现代码与解析:

双指针

原理思路:


18. 四数之和

题目描述:

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

  • 0 <= a, b, c, d < n
  • abc 和 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

实现代码与解析:

双指针

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

        for (int a = 0; a < nums.size(); a++ )
        {
            if (a > 0 && nums[a] == nums[a - 1]) continue; // a 去重
            if (nums[a] > target && nums[a] > 0) break; // 后面不可能再有符合条件的,跳过
            for (int b = a + 1; b < nums.size(); b++)
            {
                if(b > a + 1 && nums[b] == nums[b - 1]) continue; // b 去重
                if (nums[a] + nums[b] > target && nums[a] + nums[b] > 0) break; // 后面不可能再有符合条件的,跳过
                int l = b + 1, r = nums.size() - 1;
                while (l < r)
                {
                    long long sum = (long long)nums[a] + nums[b] + nums[l] + nums[r];
                    if (sum > target) r--;
                    else if (sum < target) l++;
                    else 
                    {
                        res.push_back({nums[a], nums[b], nums[l], nums[r]});
                        while(l < r && nums[l] == nums[l + 1]) l++; // l 去重
                        while(l < r && nums[r] == nums[r - 1]) r--; // r 去重
                        l++;
                        r--;
                    }
                }
            }
        }
        return res;
    }
};

原理思路:

        与Leetcode每日一题:15. 三数之和(2023.7.9 C++)_Cosmoshhhyyy的博客-CSDN博客

        一模一样,只不过这里变成了四数之和。在continue去重逻辑上与上题一样,在break剪枝的逻辑上,我们不仅要num和要大于target,同时num和大于0才能剪枝,因为target为负数的话,例如{-2, -1, 3},target = -3;,遍历-2时,虽然-2 > -3,但后面加上-1仍然可以组成-3是一种答案。

        还有注意此题注意开long long 即可。

你可能感兴趣的:(Leetcode,leetcode,c++,算法,双指针)