哈希表——6.四数相加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 。

示例：

输入：A = [1,2]
      B = [-2,-1]
      C = [-1,2]
      D = [0,2]
输出：2

与前面的三数之和、四数之和这两题不一样的是，前面这两题是在同一个数组中找3个或4个数的和等于目标值，需要考虑数重复的情况，虽然放在哈希表这部分，但并不适合用哈希法。

但本道题目是四个独立的数组，不需要考虑重复值。因此，在以后遇到题目时，如果要求你在一个数组内找几个数的和等于目标值，大概率你就要考虑用循环加双指针的方法，在过程中还需要考虑排除重复值；而在不同的数组中各找一个数，使得数之和等于目标值则可以考虑用哈希表的方法。

那么，我们直接通过代码来分析整个的解题过程，完整代码如下：

class Solution(object):
    def fourSumCount(self, nums1, nums2, nums3, nums4):
        # 使用字典存储nums1和nums2中的元素及其和
        hashmap = dict()
        for n1 in nums1:
            for n2 in nums2:
                if n1 + n2 in hashmap:
                    hashmap[n1+n2] += 1
                else:
                    hashmap[n1+n2] = 1
        
        # 如果 -(n1+n2) 存在于nums3和nums4, 存入结果
        count = 0
        for n3 in nums3:
            for n4 in nums4:
                key = - n3 - n4
                if key in hashmap:
                    count += hashmap[key]
        return count

首先，定义了一个空的字典hashmap。

将4个数组各找一个数求和等于目标值的任务分解为两个部分，一个任务是求前两个数组不同数之和，将其放在空字典中，key为两数之和，value为这两数之和出现的次数。另外一个任务是在后两个数组中求-n3-n4，判断其是否在之前的那个字典中存在，如果存在就相当于4数之和等于0。

        for n1 in nums1:
            for n2 in nums2:
                if n1 + n2 in hashmap:
                    hashmap[n1+n2] += 1
                else:
                    hashmap[n1+n2] = 1

构建两层循环，找到所有n1+n2的和，如果这个和已存在，那就将该和对应的value加1；如果不存在就新建一个键值对。

        count = 0
        for n3 in nums3:
            for n4 in nums4:
                key = - n3 - n4
                if key in hashmap:
                    count += hashmap[key]

先定义记数count = 0。然后构建两层循环，找到所有的-n3-n4，判断其是否在hashmap中有对应的key，如果有的话count加上该key对应的value，靠这种方法找到满足要求的所有组合次数。

最后return到count，找到4个数组中有几种4数之和等于0的可能。