4SUM算法

在国庆浪了一大波之后，要补的东西好像很多，所以还是先选择了一般难度的题......

先来看题

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

这道题，就是让我们在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个循环，1个循环代表一个数，然后4个数加起来，看看是否等于target...果不其然，直接给了我个超时...（我眉头一皱，发现事情并没有那么简单

所以呢，得优化一下。

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

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

tip：

因为输出不能重复，所以当前数和其前一位数相等的话，就直接continue循环了，不然会产生两个或者多个相同的结果。