leetcode 18. 4Sum

传送门

18. 4Sum

 

QuestionEditorial Solution

 My Submissions

 

• Total Accepted: 82006
• Total Submissions: 333565
• Difficulty: Medium

 

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

 

Subscribe to see which companies asked this question

Hide Tags

 Array Hash Table Two Pointers

Show Similar Problems

 

 

题意以及题解转自：

https://yq.aliyun.com/articles/3883

【题意】

 

给定n个整数的数组S，是否在 数组S中有元素a，b，c，d，使得a + b + c + d = target？在数组中找出独一无二的四元素组，使得他们之和为target。

注意：

在四元素组（a，b，c，d）中，必须满足非递减排序。 （即a≤b≤c≤d）

该解决方案集中一定不能包含重复的四元素组。

【分析】

 

1. 对数组排序
2. 确定四元数中的前两个（a，b）
3. 遍历剩余数组确定两外两个（c，d），确定cd时思路跟3Sum确定后两个数据一样，二分查找左右逼近。
4. 在去重时采用set集合

 

class Solution {
public:
    vector<vector<int>> fourSum(vector<int>& nums, int target) {
        vector<vector<int> > ans;
        int sz = nums.size();
        int i,j,sum2,te;
        int left,right;
        set<vector<int> >s;
        vector<int> tmp;
        sort(nums.begin(),nums.end() );
        for(i = 0;i < sz - 3;i++) {
            if(i != 0 && nums[i] == nums[i - 1]) continue;
            for(j = i + 1;j < sz - 2;j++) {
                if(j != i + 1 && nums[j] == nums[j - 1]) continue;
                sum2 = nums[i] + nums[j];
                left = j + 1;right = sz - 1;
                while(left < right) {
                    te = sum2 + nums[left] + nums[right];
                    if(te == target) {
                        tmp.clear();
                        tmp.push_back(nums[i]);tmp.push_back(nums[j]);tmp.push_back(nums[left]);tmp.push_back(nums[right]);
                        s.insert(tmp);
                        left++;right--;
                    }
                    else if(te < target) {
                        left++;
                    }
                    else{
                        right--;
                    }
                }
            }
        }
        set< vector<int> >::iterator it;
        for(it = s.begin();it != s.end();it++) {
            ans.push_back(*it);
        }
        return ans;
    }
};