四数相加2-哈希表

原创已于 2022-06-23 09:30:21 修改 · 222 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#散列表 #数据结构 #算法

于 2022-06-23 09:27:48 首次发布

算法专栏收录该内容

81 篇文章

订阅专栏

本文介绍了一种解决力扣上的两数相加问题的算法，通过建立哈希表记录数组A和B中元素之和的出现次数，然后遍历数组C和D，查找与C+D之和相反数在哈希表中的出现次数，累加计数，最终返回计数值。这种方法避免了重复元素的四数之和情况。

两数相加2-力扣

思路

4个独立数组，A[i] + B[j] + C[k] + D[l] = 0，不用考虑有重复的4个元素相加等于0的情况
python中，if i in dict，这里dict等价于dict.keys(),表示元素是否出现在字典的keys中
-解题步骤：

用dict做哈希表，key为a+b，value为a+b出现次数
用count统计次数，遍历数组C和D，若0-(c+d)出现在hash_map中，取出dict中对应的value，加给count
返回count

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        #用dic来做hash表，a+b作为key，a+b出现次数作为value
        hash_map=dict()        
        for a in nums1:
            for b in nums2:
                if (a+b) in hash_map:
                    hash_map[a+b]+=1
                else:
                    hash_map[a+b]=1
        
        #遍历C和D，若0-(c+d)出现在hash_map中，把对应的value取出，覆盖计数变量count
        count=0
        for c in nums3:
            for d in nums4:
                tmp=0-(c+d)
                if tmp in hash_map:
                    count+=hash_map[tmp]

        return count