代码随想录 -- 哈希表 -- 四数相加 II

454. 四数相加 II - 力扣（LeetCode）

class Solution(object):def fourSumCount(self, nums1, nums2, nums3, nums4):dic1 = {}dic2 = {}for i in nums1:for j in nums2:sum=i+jif sum in dic1:dic1[sum] += 1else:dic1[sum]=1for i in nums3:for j in nums4:sum=i+jif sum in dic2:dic2[sum] += 1else:dic2[sum]=1print dic1,dic2count = 0for i in dic1:if -i in dic2:count = count+dic2[-i]*dic1[i]return count

题解：

思路：先将 nums1 和 nums2 的和以及和出现的次数放在 hashmap 中，再计算 -(nums3 + nums4)，看其在 hashmap 中是否存在，存在就计入 count 中。

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

hashmap.get(n1+n2, 0) ：在 hashmap 中查找键 n1+n2 对应的值。如果找到了这个键，就返回它对应的值。如果没有找到这个键，就返回默认值 0。