哈希表之第454题.四数相加II

题目来源

https://leetcode.cn/problems/4sum-ii/

题目描述

给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ，使得 A[i] + B[j] + C[k] + D[l] = 0。
为了使问题简单化，所有的 A, B, C, D 具有相同的长度 N，且 0 ≤ N ≤ 500 。所有整数的范围在 -2^28 到 2^28 - 1 之间，最终结果不会超过 2^31 - 1 。

思路

两两一组，先遍历前两组统计这两数之和，使用map存储这两数这和与其出现的次数，
再遍历后两组统计此两数之和，找出此两数这和中与上面两数这和加起来等于0的次数

代码

def fourSumCount(nums1,nums2,nums3,nums4):
    hashmap=dict()         #存储数组1和数组2的和，以及次数
    for n1 in nums1:
        for n2 in nums2:
            if n1+n2 in hashmap:
                hashmap[n1+n2]+=1
            else:
                hashmap[n1+n2]=1

    count=0
    for n3 in nums3:
        for n4 in nums4:
            key=-(n3+n4)
            if key in hashmap:
                count+=hashmap[key]      #这里为什么不是加1，举例：与n3+n4对应的数-（n3+n4）在hashmap中可能多次出现，且是由不同的n1+n2组成的
    return count