454. 四数相加 II

最新推荐文章于 2024-10-01 20:17:55 发布

Marshal～

最新推荐文章于 2024-10-01 20:17:55 发布

阅读量101

点赞数

分类专栏：刷题文章标签：深度学习算法 leetcode

本文链接：https://blog.csdn.net/M_arshal_/article/details/123819599

版权

刷题专栏收录该内容

16 篇文章 0 订阅

订阅专栏

在这里插入图片描述
这里就是使用了collections.Counter，统计里面的每个数字出现的次数，比较方便。

class Solution:
    def fourSumCount(self, A: List[int], B: List[int], C: List[int], D: List[int]) -> int:
        countAB = collections.Counter(u + v for u in A for v in B)
        ans = 0
        for u in C:
            for v in D:
                if -u - v in countAB:
                    ans += countAB[-u - v]
        return ans