LeetCode 018 4sum

和3sum一样的意思,给一个序列和一个目标,在序列中找出不重复的四个数的和为目标的所有情况。

思路和3sum也一样,只是需要先将两个数相加,转化为3sum中的一个数,其他的就用3sum来做。

PS:开始是这样想的,但是优化没有这么变态,看了下discuss中的优化,简直了,leetcode中的优化感觉都是在很小很小的角落里优化的那种感觉,而且没有给定规定的时间复杂度,所以只能慢慢试了= =。。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int> >w;
        vector<int> temp;
        if(nums.size()<4)return w;
        temp.resize(4);
        sort(nums.begin(),nums.end());
        for(int ii=0;ii<nums.size()-3;ii++){
            if(nums[ii]+nums[ii+1]+nums[ii+2]+nums[ii+3]>target)break;//优化
            if(ii&&(nums[ii]==nums[ii-1]))continue;
            for(int jj=ii+1;jj<nums.size()-2;jj++){
                if(nums[ii]+nums[jj]+nums[jj+1]+nums[jj+2]>target)break;//优化
                if((jj>ii+1)&&nums[jj]==nums[jj-1])continue;
                int r=nums.size()-1;
                int l=jj+1;
                int ans=target-(nums[ii]+nums[jj]);
                while(l<r){
                    if(nums[l]+nums[r]==ans){
                       temp[0]=nums[ii];
                       temp[1]=nums[jj];
                       temp[2]=nums[l];
                       temp[3]=nums[r];
                       w.push_back(temp);
                       l++;
                       r--;
                       while(l<nums.size()&&nums[l]==nums[l-1])l++;//检查重复的
                       while(r>-1&&nums[r]==nums[r+1])r--;//检查重复的
                    }
                    while(nums[l]+nums[r]>ans)r--;
                    while(nums[l]+nums[r]<ans)l++;
                }
            }
        }
        return w;
    }
};



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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值