4Sum

来源:互联网 发布:淘宝买家号开店 编辑:程序博客网 时间:2024/05/29 14:19

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.

    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)

注意该问题的答案中,每个quadruplet内部可以有重复的数字,但quadruplet之间不能重复。即每个quadruplet内四个元素对应的index不能重复。本文给出两种解法。

解法一,先排序,然后用一个hashmap记录每两两个数的和和这两个数对应的index(注意是index而不能仅仅记录这两个数,防止重复)。时间复杂度,平均O(n^2),最坏O(n^4),空间O(n^2)。最后对结果sort并去重。代码如下。

class Solution {public:    vector<vector<int> > fourSum(vector<int> &num, int target) {        vector<vector<int> > result;        const int len = num.size();        if(len < 4) return result;        sort(num.begin(), num.end());                unordered_map<int, vector<pair<int, int> > > cache;                for(int i=0; i<len; i++)            for(int j=i+1; j<len; j++)                cache[num[i]+num[j]].push_back(pair<int,int>(i, j));                for(int i=0; i<len; ++i) {            for(int j=i+1; j<len; ++j) {                int left = target-num[i]-num[j];                if(cache.find(left) == cache.end()) continue;                                vector<pair<int,int> > t = cache[left];                for(int k=0; k<t.size(); ++k)                {                    if(i <= t[k].second)                        continue;                    result.push_back({num[t[k].first], num[t[k].second], num[i], num[j]});                }            }        }        sort(result.begin(), result.end());        result.erase(unique(result.begin(), result.end()), result.end());        return result;    }};

解法二,仍然使用hash, 时间O(n^2), 空间O(n^2)。先缓存没两个数的和,然后遍历hash,根据sum2在常数时间内找出另一个sum2,然后遍历两个sum2对应的list,根据其pair组成四元组。代码如下所示。

class Solution {public:    vector<vector<int> > fourSum(vector<int> &num, int target) {        vector<vector<int> > result;        const int len = num.size();        if(len < 4) return result;        sort(num.begin(), num.end());                unordered_map<int, vector<pair<int, int> > > cache;                for(int i=0; i<len; i++)            for(int j=i+1; j<len; j++)                cache[num[i]+num[j]].push_back(pair<int,int>(i, j));                unordered_map<int, vector<pair<int, int> > >::iterator it = cache.begin();        for(; it != cache.end(); ++it)        {            int left = target-it->first;            if(cache.find(left) == cache.end()) continue;                        vector<pair<int, int> > v1 = it->second;            vector<pair<int, int> > v2 = cache[left];                for(int i=0; i<v1.size(); ++i)                for(int j=0; j<v2.size(); ++j)                {                    auto a = v1[i].first;                    auto b = v1[i].second;                    auto c = v2[j].first;                    auto d = v2[j].second;                    if(b < c)                        result.push_back({num[a], num[b], num[c], num[d]});                }        }                sort(result.begin(), result.end());        result.erase(unique(result.begin(), result.end()), result.end());        return result;    }};

参考求和问题总结(leetcode 2Sum, 3Sum, 4Sum, K Sum)):http://tech-wonderland.net/blog/summary-of-ksum-problems.html


0 0