4Sum算法

国庆浪了一大段时间,现在得好好补补了......

由于要补的东西比较多,所以还是选择了一般难度的题

好了,言归正传,先看题

Given an array S of n integers, are there elements abc, and d in S such that a + b + c + d = target? Find all unique quadruplets in the array which gives the sum of target.

For example, given array S = [1, 0, -1, 0, -2, 2], and target = 0.

A solution set is:
[
  [-1,  0, 0, 1],
  [-2, -1, 1, 2],
  [-2,  0, 0, 2]
]
大致理解一下,就是给出一个target,然后从S中找出4个数,加起来等于这个target,最后的结果由小到大排序。

先看代码吧

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
         vector<vector<int>> ans;
        int n = nums.size();
        if(n < 4)  return ans;
        sort(nums.begin(),nums.end());
        for(int i = 0; i < n-3; i++)
        {
            if(i > 0&&nums[i] == nums[i - 1]) continue;
        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;
        for(int j = i + 1;j < n - 2;j++)
        {
            if(j > i + 1&&nums[j] == nums[j - 1]) continue;
            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;
            for(int p = j + 1; p < n - 1; p++){
                if(p > j + 1&&nums[p] == nums[p - 1]) continue;
                if(nums[i] + nums[j] + nums[p] + nums[p + 1] > target) break;
                if(nums[i] + nums[j] + nums[p] + nums[n - 1] < target) continue;
                for(int q = p + 1; q < n; q++){
                    if(q > p + 1&&nums[q] == nums[q - 1]) continue;
                        
                    int sum = nums[i] + nums[j] + nums[p] + nums[q];
                    if(target == sum){
                        ans.push_back(vector<int>{ nums[i], nums[j], nums[p], nums[q]});
                    }
                }
            }
        }
        }
        return ans;
    }
};


大致想法就是,因为输出要由小到大嘛,那就干脆直接先排个序,再来分析。

我们先选第一个数,这个数是从第一位从小到大一个一个选的,第二个数则要从第一个数的下一位开始一个一个选,以此类推。然后4个循环,把四个数加起来,看等不等于target就行了。但是,结果并没有那么简单,系统直接给了我个超时,真棒......

所以,得优化一下,免去一些不必要的操作。

第一,在当前数的循环中,把前面的数、当前数和当前数的后几位(凑够4位)加起来,如果结果比target大,那么直接跳出当前数的循环。好像有点难懂,举个例子吧,比如说现在在第二层循环,即当前数是第二个数,就把第一个数,第二个数和第二个数的后两位数加起来,看结果是不是比target大。因为这个结果是取当前数的情况下的最小和了,如果最小和都比target大,那么当前数和之后的数就都没有意义了,直接break跳出当前数循环。

第二,在当前数的循环中,把前面的数、当前数和最后的几位(凑够4位)加起来,如果结果比target小,那么直接在当前数的循环实行continue。举个例子吧,比如说现在在第二层循环,即当前数是第二个数,就把第一个数,第二个数和倒数后两位数加起来,看结果是不是比target小。因为这个结果是取当前数的情况下的最大和了,如果最大和都比target小,那么当前数就没有意义了,需要更大的当前数,直接continue。


tip:

排完序之后,数值相同的数会黏在一起,当遇到当前数字与前一个相同时,就要直接跳过,不然会得出两个相同的结果。


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值