题目:
算法:
哈希法。当我们需要查询一个元素是否出现过,或者一个元素是否在集合里的时候,就要第一时间想到哈希法。
类似两数相加
比如四个数组为A B C D
可把A和B看成一组,C和D看成一组。
A和B遍历完以后存放在字典里面,key是a+b,value是a+b出现的次数
被加数x=0-(C+D)
查询x是否在字典里面(哈希),并查询x出现的次数。
调试过程:
class Solution:
def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
#遍历nums1,nums2.将其和、出现次数存储在dict中
dict = {}
count = 0
for a in nums1:
for b in nums2:
if a+b in dict:
dict[a+b] += 1
else:
dict[a+b] = 1
for c in nums3:
for d in nums4:
x = 0-(c+d)
if x in dict:
count +=1
return count
原因:cd循环中计数应该是:count += dict[x]
count统计a+b+c+d=0的组合的个数。当x in dict时,此时已经有对应个数的组合数了。
比如:a[0]+b[0]=5, a[1]+b[1]=5,则dict中dict[5]=2
此时,c[0]+d[0]=-5,那么循环到c[0]+d[0]时,count应该等于2
当cd的下一轮循环中,c[1]+d[1]=-5,此时c[1]+d[1]+a[0]+b[0]=5,c[1]+d[1]+a[1]+b[1]=5,增加了两个组合,count应该继续加2
修改后得到正确代码:
class Solution:
def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
#遍历nums1,nums2.将其和、出现次数存储在dict中
dict = {}
count = 0
for a in nums1:
for b in nums2:
if a+b in dict:
dict[a+b] += 1
else:
dict[a+b] = 1
for c in nums3:
for d in nums4:
x = 0-(c+d)
if x in dict:
count += dict[x]
return count
-
时间复杂度: O(n^2)
-
空间复杂度: O(n^2),最坏情况下A和B的值各不相同,相加产生的数字个数为 n^2