454. 四数相加 II/C++

最新推荐文章于 2024-10-14 19:42:49 发布

原创最新推荐文章于 2024-10-14 19:42:49 发布 · 299 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode/C++ 专栏收录该内容

248 篇文章

订阅专栏

本文介绍了一种高效计算四个整数数组中是否存在两组数相加之和为零的方法。通过使用哈希表提前存储两个数组的所有可能组合，再遍历另外两个数组查找相反数，实现了快速查找。

在这里插入图片描述
先将A与B的组合存入hash，如果C与D的组合是hash元素的相反数，那么次数就加对应元素的个数。

int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
    int ans = 0;
    unordered_map<int,int> ab;
    for(auto a : A)
        for(auto b : B)
            ++ab[a + b];
        
    for(auto c : C)
        for(auto d : D){
            int need = -(c + d);
            if(ab.count(need))  ans += ab[need];
        }
    
    return ans;
}