LeetCode--4Sum II（4个数的和）Python

最新推荐文章于 2022-03-29 20:45:26 发布

原创最新推荐文章于 2022-03-29 20:45:26 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

101 篇文章

订阅专栏

本文介绍了一种高效解决四数组求和等于零的问题的方法。通过使用哈希表存储两个数组的所有可能组合之和，然后查找剩余两个数组的和是否为前者的负数，从而快速得出所有满足条件的组合数量。

题目：

给定四个长度为N的数组：A、B、C、D，找到A[i]+B[j]+C[k]+D[l]=0的tuple个数。

解题思路：

直接对四个数组进行遍历，时间复杂度为N的4次方，超时。之后想到将哈希应用到计算过程中。先将其中两个数组的和存储到字典中，再计算另外两个数组的和的相反数有没有在之前的字典中出现，出现则将最终的count数加上其出现的次数。时间复杂度为n的2次方。

代码（Python）：

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
        """
        count = 0
        n = len(A)
        Dict = {}
        for i in range(n):
            for j in range(n):
                temp = A[i]+B[j]
                if temp in Dict:
                    Dict[temp] = Dict[temp]+1
                else:
                    Dict[temp] = 1
        for i in range(n):
            for j in range(n):
                temp = -(C[i]+D[j])
                if temp in Dict:
                    count = count + Dict[temp]
                else:
                    continue
        return count