python leetcode 454. 4Sum II

Neekity

于 2018-12-03 09:18:16 发布

阅读量613

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode python 文章标签： python leetcode

本文链接：https://blog.youkuaiyun.com/Neekity/article/details/84727076

python 同时被 2 个专栏收录

263 篇文章

订阅专栏

leetcode

221 篇文章

订阅专栏

本文介绍了一种使用字典解决四数之和计数问题的方法，通过预先计算两个数组的所有可能组合，并将结果存储在字典中，然后遍历另外两个数组，查找与当前元素相加为零的配对，从而实现快速计数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这类题目基本是用字典以空间换时间

class Solution(object):
    def fourSumCount(self, A, B, C, D):
        """
        :type A: List[int]
        :type B: List[int]
        :type C: List[int]
        :type D: List[int]
        :rtype: int
        """
        dict1={}
        res=0
        for i in range(len(A)):
            for j in range(len(B)):
                if not A[i]+B[j] in dict1:
                    dict1[A[i]+B[j]] = 1
                else:
                    dict1[A[i]+B[j]]+=1
        for m in range(len(C)):
            for n in range(len(D)):
                if -(C[m]+D[n]) in dict1:
                    res+=dict1[-(C[m]+D[n])]
        return res