Leetcode454.四数相加（哈希表分组 C++）

最新推荐文章于 2024-05-20 15:46:19 发布

小码大哈

最新推荐文章于 2024-05-20 15:46:19 发布

阅读量596

点赞数

分类专栏： leetcode 哈希表文章标签： leetcode 数据结构算法

本文链接：https://blog.csdn.net/RuiZzi/article/details/123646459

版权

leetcode 同时被 2 个专栏收录

10 篇文章 0 订阅

订阅专栏

哈希表

5 篇文章 0 订阅

订阅专栏

Leetcode454.四数相加（哈希表分组 C++）

题目描述：

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

示例

在这里插入图片描述

知识点

哈希表+分组

思路

先定义一个unordered_map，key存放的是a+b，value存放a+b的个数。
再定义一个count = 0 ，作为最后的输出结果。
再对c和d进行遍历，在哈希表中寻找符合条件的情况，如果符合则count += map[-c-d]
最后返回count

代码

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int, int>umap;
        for(int a : nums1)
            for(int b : nums2)
            {
                umap[a+b]++;
            }
        int count = 0;
        for(int c : nums3)
            for(int d : nums4)
            {
                if(umap.find(0-(c+d)) != umap.end())
                {
                    count +=umap[0-(c+d)];
                }
            }
        return count;
    }
};