LeetCode刷题之454. 四数相加 II

wuchangjian2021-10-25 03:00:15编程学习

LeetCode刷题之454. 四数相加 II

  • 题目

给你四个整数数组 nums1nums2nums3nums4 ,数组长度都是 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 ≤ n u m s 1 [ i ] , n u m s 2 [ i ] , n u m s 3 [ i ] , n u m s 4 [ i ] ≤ 2 28 -2^{28}\le nums1[i], nums2[i], nums3[i], nums4[i] \le 2^{28} 228nums1[i],nums2[i],nums3[i],nums4[i]228
  • 代码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
  • 算法说明:
    遍历nums1nums2中元素和的组合情况,并记录在字典temp中,temp的键值 key 为两数和,value 为该两数和出现的次数。
    注意: 字典操作dict.get(key, default=None),返回指定键的值,如果值不在字典中,返回default值;
    遍历nums3nums4中元素和的组合情况,取和s的负值判断其是否在temp中,若存在,则取出temp对应的 value 值temp[s],更新result的值,result += temp[s]
    注意: 如果和为0,则不取符号,不然报错。

胡萝卜

2021年10月24日18:27:42

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

发表评论    

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。