3.5四数相加(LC454-M)

题目:

算法:

哈希法。当我们需要查询一个元素是否出现过,或者一个元素是否在集合里的时候,就要第一时间想到哈希法。

类似两数相加

比如四个数组为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

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值