哈希：四数相加

最新推荐文章于 2024-05-25 14:25:44 发布

代码拌饭饭更香

最新推荐文章于 2024-05-25 14:25:44 发布

阅读量167

点赞数 1

分类专栏：数据结构与算法

本文链接：https://blog.csdn.net/cckluv/article/details/113825006

版权

哈希表 unordered_map 四数之和计数数据结构优化

关键词由CSDN通过智能技术生成

数据结构与算法专栏收录该内容

57 篇文章 2 订阅

订阅专栏

哈希：四数相加

在这里插入图片描述

首先定义一个unordered_map，key放a和b两数之和，value 放a和b两数之和出现的次数。
遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。
定义int变量count，用来统计a+b+c+d = 0 出现的次数。
在遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。
最后返回统计值 count 就可以了

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int, int> map;
        for(auto &a : A)
        {
            for(auto &b : B)
            {
                map[a + b]++;
            }
        }
        int count = 0;
        for(auto &c : C)
        {
            for(auto &d : D)
            {
                if(map.find(0 - (c + d)) != map.end())
                {
                    count += map[-(c + d)];
                }
            }
        }
        return count;
    }
};