代码随想录算法训练营第七天|454.四数相加II

最新推荐文章于 2026-02-06 18:32:23 发布

原创最新推荐文章于 2026-02-06 18:32:23 发布 · 191 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

收录于

当前文章被以下社区和专栏收录：

题目链接/文章讲解/视频讲解：https://programmercarl.com/0454.%E5%9B%9B%E6%95%B0%E7%9B%B8%E5%8A%A0II.html

这题的解题过程，直接用四个for循环去暴力解的话时间复杂度太高，通过将前两个数组合并（各个数相加）成一个字典，后两个数组合并（各个数相加）成一个字典从而降低时间复杂度。为什么合并成字典，是因为合并之后出现的相同数字要算上次数。在此不熟悉的可以查一下数组和字典的区别。

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

标签