4Sum(middle)

最新推荐文章于 2024-09-16 16:07:55 发布

YT的呼呼

最新推荐文章于 2024-09-16 16:07:55 发布

阅读量189

点赞数

分类专栏：算法文章标签： c++ 算法刷题 leetcode 4Sum

本文链接：https://blog.csdn.net/u012345366/article/details/54799602

版权

算法专栏收录该内容

11 篇文章 0 订阅

订阅专栏

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

Note: The solution set must not contain duplicate quadruplets.

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

需要先缩小范围，先将数组排序，我们可以知道排序完的数组是从小到大排列，如果每层循环不限制跳出条件，时间就会超出。从代码中可以看到缩小的范围。

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int>> result;
        int num=nums.size();
        sort(nums.begin(),nums.end());
        for(int i=0;i<num-3;i++){
            if(i>0&&nums[i]==nums[i-1])continue;//元素不能相同，不然放入vector中就会出现重复
            if(nums[i]+nums[i+1]+nums[i+2]+nums[i+3]>target)break;//找到数组中四个最小的，如果这四个最小之和也大于target，那么跳出循环
            if(nums[i]+nums[num-3]+nums[num-2]+nums[num-1]<target) continue;//如果这个nums[i]与最大的三个数之和还小于target，找下一个
            for(int j=i+1;j<num-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[num-2]+nums[num-1]<target)continue;
                int left=j+1;
                int right=num-1;
                while(left<right){
                    if(left>j+1&&nums[left]==nums[left-1]||nums[i]+nums[j]+nums[left]+nums[right]<target){
                        left++;
                        continue;
                    }
                    if(right<num-1&&nums[right]==nums[right+1]||nums[i]+nums[j]+nums[left]+nums[right]>target){
                        right--;
                        continue;
                    }
                    if(nums[i]+nums[j]+nums[left]+nums[right]==target){
                        vector<int>tmp;
                        tmp.push_back(nums[i]);
                        tmp.push_back(nums[j]);
                        tmp.push_back(nums[left]);
                        tmp.push_back(nums[right]);
                        result.push_back(tmp);
                        left++;
                        right--;
                    }
                }
                
            }
        }
        return result;
    }
};