四数相加- 分组+HASH

Problem: 454. 四数相加 II

思路

分组 + hash表：先将4个数组两两划分，分别计算两组中两个数组各取一个元素的和作为这两组的元素(时间复杂度O(n))，由于只需要计算总和为零的个数，所以可以以hash表存储各组的两数组取数元素和，和该元素和的出现次数。此时问题简单成，两个hash表中各取一个元素其和为0的出现次数；这可以通过遍历其中一个hash表的方式实现

解题方法

dicA是存储nums1和nums2中各取一个元素求和，元素和和其出现次数的hash表，dicB是存储nums3和nums4中各取一个元素求和，元素和和其出现次数的hash表。通过两层的遍历分别实现这两个字典，再遍历dicA的键值k, v，若-k出现在dicB中，表示nums1和nums2组成k的元素加上nums3和nums4组成-k的元素求和，值等于0，将出现次数(即dicA次数*dicB次数)累加到res中。

复杂度

时间复杂度:

$O(n^2)$

空间复杂度:

$O(n^2)$

Code

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        dicA = {}
        dicB = {}
        for n1 in nums1:
            for n2 in nums2:
                dicA[n1+n2] = dicA.get(n1+n2, 0) + 1
            
        for n3 in nums3:
            for n4 in nums4:
                dicB[n3 + n4] = dicB.get(n3+n4, 0) + 1
        
        res = 0
        for k, v in dicA.items():
            if dicB.get(-k) is not None:
                res += v * dicB[-k]
        
        return res