数组 4Sum

思想:

方法一:

先排序再左右夹逼,时间复杂度O(n^3),但会TLE。

//4Sum O(n^3)
vector<vector<int>> fourSum(vector<int> &num, int target) {
    vector<vector<int>> result;
    if(num.size()<4) return result;
    sort(num.begin(),num.end());
    auto last = num.end();
    for(auto a = num.begin(); a < prev(last,3); a++) {
        for(auto b = next(a); b < prev(last,2); b++) {
            auto c = next(b);
            auto d = prev(last);
            while(c<d) {
                const int sum = *a+*b+*c+*d;
                if(sum<target) {
                    c++;
                }else if(sum>target) {
                    d--;
                }else {
                    result.push_back({*a,*b,*c,*d});
                    ++c;
                    --d;
                }
            }
        }
    }
    sort(result.begin(),result.end());
    result.erase(unique(result.begin(), result.end()), result.end());
    return result;
}

方法二:

先用hashmap缓存两两数的和:key为和,value为下标对<a,b>。

然后找另外两个数(下标为c,d),满足总和为0:

如果找到下标为a,b,c,d顺序的四数保留(a,b,c,d互不相等)

最后去重。

时间复杂度:O(n^2)

class Solution {
public:
    vector<vector<int> > fourSum(vector<int> &num, int target) {
        vector<vector<int>> result;
        if(num.size() < 4) 
            return result;
        sort(num.begin(),num.end());
        
        unordered_map<int,vector<pair<int,int>>> cache;
        
        for(size_t a = 0; a < num.size(); ++a) {
            for(size_t b = a+1; b < num.size(); ++b) {
                cache[num[a]+num[b]].push_back(pair<int,int>(a,b));
            }
        }
        for(size_t c = 0; c < num.size(); ++c) {
            for(size_t d = c+1; d < num.size(); ++d) {
                const int key = target - num[c] - num[d];
                if(cache.find(key) == cache.end())
                    continue;
                const auto &vec = cache[key];
                for(size_t k = 0; k < vec.size(); ++k) {
                    if(c <= vec[k].second) 
                        continue;
                    result.push_back({num[vec[k].first], num[vec[k].second], num[c], num[d]});
                }
            }
        }
        sort(result.begin(), result.end());
        result.erase(unique(result.begin(),result.end()),result.end());
        return result;
    }
};




  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值