Leetcode 454. 四数相加 II 4Sum II - Python 哈希法

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        hash = dict()
#统计nums1和nums2所有potential sum 的次数,将其放到哈希表中
        for n1 in nums1:
            for n2 in nums2:
                if n1 + n2 in hash:
                    hash[n1+n2] += 1
                else:
                    hash[n1+n2] = 1
 #由a+b+c+d=0 -> a+b = 0-(c+d) = key
 #若key存在于hash表中,则就说明此等式成立,key对应的value就是等式成立的次数
        count = 0
        for n3 in nums3:
            for n4 in nums4:
                key = 0 - (n3 + n4)
                if key in hash:
                    count += hash[key]
        return count

将nums1和nums2之和看成一组,将nums3和nums4之和看成一组。

然后用leetcode第1题的思路:如果target - (c+d)在hash之中,则符合条件。

你可能感兴趣的:(leetcode,哈希表,力扣,哈希算法,leetcode,python)