454. 四数相加 II（unordered_map）

最新推荐文章于 2025-02-10 23:30:27 发布

原创最新推荐文章于 2025-02-10 23:30:27 发布 · 156 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #数据结构 #哈希

leetcode 专栏收录该内容

71 篇文章

订阅专栏

该博客介绍了一个C++实现的算法，用于计算四个整数数组nums1、nums2、nums3和nums4中，任意两数组之和等于另外两数组之和的组合数量。算法利用哈希表存储前两个数组的和及其出现次数，然后查找后两个数组之和的相反数在哈希表中出现的次数，最后返回累加的计数结果。

题目描述

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        // nums1 与 nums2 的和作为key，出现的次数作为 value
        unordered_map<int, int> sum12;
        for (auto i : nums1) {
            for (auto j : nums2) {
                sum12[i + j]++;
            }
        }
        // 0 - (nums3 + nums4) 在sum12中是否出现，并将次数累加
        int count = 0;
        for (auto i : nums3) {
            for (auto j : nums4) {
                if (sum12.find(0 - (i + j)) != sum12.end()) {
                    count += sum12[0 - (i + j)];
                }
            }
        }
        return count;
    }
};