454. 四数相加 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 。所有整数的范围在 -228 到 228 - 1 之间，最终结果不会超过 231 - 1 。

原题请参考链接https://leetcode-cn.com/problems/4sum-ii/

题解

方法一 【哈希表(哈希Map)】

class Solution:
    def fourSumCount(self, A: List[int], B: List[int], C: List[int], D: List[int]) -> int:
        res = 0
        count_ab = {}
        # 数组A和数组B的所有和放到Map中
        for i in A:
            for j in B:
                tmp = i + j
                if tmp in count_ab:
                    # 计算出现的次数
                    count_ab[tmp] += 1
                else:
                    count_ab[tmp] = 1
        # 数组C和数组D的所有和与Map对比，相同则满足要求
        for k in C:
            for l in D:
                if -(k+l) in count_ab:
                    res += count_ab[-(k+l)]
        return res