18. 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.

Note:

  • Elements in a quadruplet (a,b,c,d) must be in non-descending order. (ie, a ≤ b ≤ c ≤ d)
  • The solution set must not contain duplicate quadruplets.

注意:

加入剪枝条件:

if(nums[i]+nums[j]+nums[j+1]+nums[j+2]>target)
  break;
if(nums[i]+nums[j]+nums[l-2]+nums[l-1]<target)
  continue;
加之前120ms,之后20ms。
class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        int l = nums.size(), sum, left, right, i, j;
        vector<vector<int>> ans;
        if (l<4)
            return ans;
        sort(nums.begin(), nums.end());
        for (i = 0; i<l - 3; i++)
        {
            if(i && nums[i] == nums[i-1])
                continue;
            for (j = i + 1; j<l - 2; j++)
            {
                if(nums[i]+nums[j]+nums[j+1]+nums[j+2]>target)
                    break;
                if(nums[i]+nums[j]+nums[l-2]+nums[l-1]<target)
                    continue;
                if (j>i + 1 && nums[j] == nums[j - 1])
                    continue;
                sum = nums[i] + nums[j];
                left = j + 1;
                right = l - 1;
                while (left < right)
                {
                    if (sum + nums[left] + nums[right] == target)
                        ans.push_back({ nums[i], nums[j], nums[left++], nums[right--] });
                    else if (sum + nums[left] + nums[right] < target)
                        left++;
                    else
                        right--;
                    while (left>j+1 && nums[left] == nums[left - 1])
                        left++;
                    while(right<l-1 && nums[right] == nums[right + 1])
                        right--;
                }
            }
        }
        return ans;
    }
};
 
原文地址:https://www.cnblogs.com/argenbarbie/p/5224302.html