leetcode四数之和

1.递归+回溯

需要考虑数组中可能存在重复的元素,故先对数组进行排序

遍历数组,当前元素与其前的元素值相同时,则跳过

代码如下:

class Solution {
    vector<vector<int>> res;
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
         int len=nums.size();
         vector<int> ans;
         sort(nums.begin(),nums.end());
         for(int i=0;i<len-3;i++)
         {
             if(i>0&&nums[i]==nums[i-1])
               continue;
             ans.push_back(nums[i]);
             dfs(nums,ans,i+1,target);
             ans.pop_back();
         }
         return res;
    }
    void dfs(vector<int> &nums,vector<int> &ans,int start,int target)
    {
        if(ans.size()==4)
        {
            if(ans[0]+ans[1]+ans[2]+ans[3]==target)
            {
                res.push_back(ans);
                return;
            }
        }
        for(int i=start;i<nums.size();i++)
        {
            if(i>start&&nums[i]==nums[i-1])
              continue;
            ans.push_back(nums[i]);
            dfs(nums,ans,i+1,target);
            ans.pop_back();
        }
    }
};

效率太低,结果超时

2.双层遍历+双指针

先排序,双层遍历之后使用双指针

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        sort(nums.begin(),nums.end());
        vector<vector<int>> result;
        int size=nums.size();
        for(int a=0;a<size-3;++a)
        {
            if(a>0&&nums[a]==nums[a-1])continue;
            for(int b=a+1;b<size-2;++b)//以下代码与三数之和一样的
            {
                if(b>a+1&&nums[b]==nums[b-1])continue;
                int i=b+1,j=size-1;
                while(i<j)
                {
                    int sum=nums[a]+nums[b]+nums[i]+nums[j];
                    if(sum<target)
                        while(i<j&&nums[i]==nums[++i]);
                    else if(sum>target)
                        while(i<j&&nums[j]==nums[--j]);
                    else{
                        result.push_back(vector<int>{nums[a],nums[b],nums[i],nums[j]});
                        while(i<j&&nums[i]==nums[++i]);
                        while(i<j&&nums[j]==nums[--j]);
                    }
                }
            }
        }
        return result;
    }
};

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值