leetcode 454 四数相加

最新推荐文章于 2024-07-23 18:01:36 发布

weixin_34162695

最新推荐文章于 2024-07-23 18:01:36 发布

阅读量67

点赞数

原文链接：http://www.cnblogs.com/joelwang/p/10984232.html

版权

采用一个哈希表存储两个数的和，再遍历另外两个数组的和，time O(n2) space O(n2)

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {

        int N=A.size();
        if(N==0) return 0;
        int res=0;
        unordered_map<int,int> h;
        for(auto a:A){
            for(auto b:B){
                h[a+b]++;
            }
        }
        for(auto c:C){
            for(auto d:D){
                auto it=h.find(-c-d);
                if(it!=h.end()) res+=it->second;
            }
        }
        return res;
    }
};