LeetCode刷题之454. 四数相加 II_给你四个整数数组,nums1,nums2,nums3和nums4 python解法-CSDN博客

本文链接：https://blog.csdn.net/qq_34331113/article/details/120937884

LeetCode刷题之454. 四数相加 II

题目：

给你四个整数数组 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
- $-2^{28}\le nums1[i], nums2[i], nums3[i], nums4[i] \le 2^{28}$
代码1：

class Solution:
    def fourSumCount(self, nums1: List[int], nums2: List[int], nums3: List[int], nums4: List[int]) -> int:
        result = 0
        temp = dict()
        for i in nums1:
            for j in nums2:
                temp[i+j] = temp.get(i+j,0) + 1
        for m in nums3:
            for n in nums4:
                if m + n == 0:
                    s = m+n
                else:
                    s = -(m+n)
                if s in temp:
                    result += temp[s]
        return result

算法说明：
① 遍历nums1和nums2中元素和的组合情况，并记录在字典temp中，temp的键值 key 为两数和，value 为该两数和出现的次数。
注意： 字典操作dict.get(key, default=None)，返回指定键的值，如果值不在字典中，返回default值；
② 遍历nums3和nums4中元素和的组合情况，取和s的负值判断其是否在temp中，若存在，则取出temp对应的 value 值temp[s]，更新result的值，result += temp[s]；
注意： 如果和为0，则不取符号，不然报错。