【力扣刷题】【1-50】18. 四数之和

18. 四数之和

1.双指针法

  • 和前面的求和题目相似,这里我们用双指针法降低算法复杂度,从O(n4)到O(n3)。4个移动指针,三重遍历,在第三重使用双指针。

  • 双指针法本身并不复杂,在这里关键是我们要做好剪枝 ,才是优秀的双指针方法。

  • 剪枝1:去除重复元素,四个指针,每次移动之后都要和前一个数值比较,避免重复元素参与比较;

  • 剪枝2:去除绝对不可能得到结果的位置:

    • 确定ind1后,如果ind1+ind1后面三个元素已经>target,那后面的结果必然也大于target,跳出循环;
      • 同理,确定ind1,ind2后,如果ind1+ind2+ind2后面两个元素已经>target,跳出;
    • 确定ind1后,如果ind1+末尾三个元素<target,那么ind1开头的所有结果,必然都小于target,这一个ind1就不遍历,直接前进。
      • 和上同理,确定ind1,ind2后,也能进行判断,无法得到结果的ind2就跳过,直接前进。
  • class Solution {
    public:
        vector<vector<int>> fourSum(vector<int>& nums, int target) {
            vector<vector<int>> result;
            int ind1,ind2,ind3,ind4,n=nums.size();
            long long now_val;
            sort(nums.begin(),nums.end());
            for(ind1=0;ind1<n-3;ind1++){
                if(ind1>0&&nums[ind1-1]==nums[ind1]){continue;}
    
                now_val=(long long)nums[ind1]+nums[ind1+1]+nums[ind1+2]+nums[ind1+3];//使用int这里有一个用例会溢出
                if(now_val>target){break;}//1.ind1+ind1后面3个数加起来已经大于 target,就再也不会有新结果了
    
                now_val=(long long)nums[ind1]+nums[n-1]+nums[n-2]+nums[n-3];
                if(now_val<target){continue;}//1.ind1+最后3个数<target,当前的ind1不符合条件
    
                for(ind2=ind1+1;ind2<n-2;ind2++){
                    if(ind2>ind1+1&&nums[ind2-1]==nums[ind2]){continue;}
    
                    now_val=(long long)nums[ind1]+nums[ind2]+nums[ind2+1]+nums[ind2+2];
                    if(now_val>target){break;}//2.ind1+ind2+ind2后面2个数加起来已经大于 target,就再也不会有新结果了
    
                    now_val=(long long)nums[ind1]+nums[ind2]+nums[n-1]+nums[n-2];
                    if(now_val<target){continue;}//2.ind1+ind2+最后2个数加起来< target,当前的ind2不符合条件
    
                    ind3=ind2+1;
                    ind4=n-1;
    
                    while(ind3<ind4){
                        now_val=nums[ind1]+nums[ind2]+nums[ind3]+nums[ind4];
                        if(now_val<target){
                            ind3++;
                            while(ind3<ind4&&nums[ind3-1]==nums[ind3]){ind3++;}
                        }else if(now_val>target){
                            ind4--;
                            while(ind4>ind3&&nums[ind4+1]==nums[ind4]){ind4--;}
                        }else{
                            vector<int> v{nums[ind1],nums[ind2],nums[ind3],nums[ind4]};
                            result.push_back(v);
                            ind3++;
                            while(ind3<ind4&&nums[ind3-1]==nums[ind3]){ind3++;}
                            ind4--;
                            while(ind4>ind3&&nums[ind4+1]==nums[ind4]){ind4--;}
                        }
                    }
                }
            }
            return result;
        }
    };
    
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值