力扣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]]

二,解

本题依然采用双指针法

具体来说,对比三数之和最快的双指针方法,我们只需在外层在嵌套一层循环即可。这样时间复杂度将从O(N^2)变为 O(N^3)

提升速度方法(判重,边界条件):

指针依次是 p,k,i,j,如果 nums[p] + 3 * nums[p + 1] > target,因为 nums 按升序排列,所以之后的数肯定都大于 target ,直接 break;

如果 nums[p] + 3 * nums[-1] < target,那么当前的 nums[p] 加其余三个数一定小于 target,故 p 直接下一位即可,continue;

k 和 p 判断完全一样,只是将 3 变成了 2,target 变成了 target - nums[p]。

同样地,为了避免结果重复,某个指针遇到相同的数需要直接跳过,这与三数之和是一样的。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> res;
        if(nums.size()<4)return res;
        size_t n=nums.size();
        sort(nums.begin(),nums.end());
        for(int i=0;i<n-3;i++){
            //边界条件,不存在
            if(target<=0&&nums[i]>0)break;
            if(nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target)break;
            //条件不满足
            if(nums[i]+nums[n-3]+nums[n-2]+nums[n-1]<target)continue;
            //去重
            if(i>0&&nums[i]==nums[i-1])continue;
            for(int j=i+1;j<n-2;j++){
                //边界条件,不存在
                if(nums[i]+nums[j]+nums[j+1]+nums[j+2]>target)break;
                //条件不满足
                if(nums[i]+nums[j]+nums[n-2]+nums[n-1]<target)continue;
                //去重
                if(j>i+1&&nums[j]==nums[j-1])continue;
                int left=j+1,right=n-1;
                while(left<right){
                    //a:nums[i],b:nums[j],c:nums[left],nums[j],求当前四元组之和,和目标target比较
                    int sum=nums[i]+nums[j]+nums[left]+nums[right];
                    if(sum<target)left++;
                    //双指针移动
                    else if(sum>target)right--;
                    else {
                        //找到对应四元组
                        res.push_back({nums[i],nums[j],nums[left],nums[right]});
                        int last_start=left,last_end=right;
                        //找到四元组后去重c,d
                        while(left<right&&nums[left]==nums[last_start])left++;
                        while(left<right&&nums[right]==nums[last_end])right--;
                    }

                }
            }
        }
        return res;

    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Nathaniel333

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值