LeetCode 454 四数相加 II

最新推荐文章于 2024-07-27 15:34:52 发布

Gordennizaicunzai

最新推荐文章于 2024-07-27 15:34:52 发布

阅读量494

点赞数

分类专栏： LeetCode 文章标签： LeetCode 454 四数相加 II hash

本文链接：https://blog.csdn.net/Gordennizaicunzai/article/details/122630679

版权

LeetCode 专栏收录该内容

34 篇文章 0 订阅

订阅专栏

454. 四数相加 II

难度中等476收藏分享切换为英文接收动态反馈

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

示例 1：

输入：nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出：2
解释：
两个元组如下：
1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0

示例 2：

输入：nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出：1

提示：

n == nums1.length
n == nums2.length
n == nums3.length
n == nums4.length
1 <= n <= 200
-228 <= nums1[i], nums2[i], nums3[i], nums4[i] <= 228

利用hash表，四数相加转变成两数之和等于目标值。

def fourSumCount(self, nums1, nums2, nums3, nums4):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :type nums3: List[int]
        :type nums4: List[int]
        :rtype: int
        """
        hash_map = dict()
        counts = 0

        for i in nums1:
            for j in nums2:
                if i + j in hash_map:
                    hash_map[i+j] += 1
                else:
                    hash_map[i+j] = 1
        
        for i in nums3:
            for j in nums4:
                key = 0 - (i + j)
                if key in hash_map:
                    counts += hash_map[key]
        
        return counts