Leetcode454. 四数相加 II

Grairain

于 2021-04-07 11:47:44 发布

阅读量75

点赞数

分类专栏： Leetcode刷题文章标签： leetcode hashmap

本文链接：https://blog.youkuaiyun.com/weixin_45845896/article/details/115216707

版权

Leetcode刷题专栏收录该内容

25 篇文章

订阅专栏

这篇博客探讨了一种优化算法，用于计算四个整数数组A、B、C和D中，任意两对元素相加等于零的组合数量。原始方法使用两个映射记录相加结果，而改进后的实现则在遍历C和D数组时直接查找目标值，减少了迭代次数，提高了效率。这种方法类似于两数之和问题的解决方案。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最初的想法：用两个map来记录相加结果

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int, int> m;
        unordered_map<int, int> n;
        for(int a : A){
            for(int b : B){
                m[a + b]++;
            }
        }

        for(int c : C){
            for(int d : D){
                n[c + d]++;
            }
        }      

        int cnt = 0;

        for(auto &item : m){
            if(n.find(0-item.first) != n.end()){
                cnt += n[0-item.first] * item.second;
            }
        }
        return cnt;
    }
};

改进写法：在第二个大循环里就可以进行判断了。

遍历大C和大D数组时，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。其思想和两数之和差不多。

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int, int> m;
        for(int a : A){
            for(int b : B){
                m[a + b]++;
            }
        }

        int cnt = 0;

        for(int c : C){
            for(int d : D){
                if(m.find(0 - c - d) != m.end()){
                    cnt += m[0 - c - d];
                }
            }
        }
        return cnt;
    }
};