LeetCode刷题之454. 四数相加 II

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，则不取符号，不然报错。

胡萝卜

2021年10月24日18:27:42

我不知道将去向何方，但我已在路上！
时光匆匆，虽未曾谋面，却相遇于斯，实在是莫大的缘分，感谢您的到访 ！