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 ≤ 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} −228≤nums1[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
- 算法说明:
① 遍历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
我不知道将去向何方,但我已在路上! |
---|
时光匆匆,虽未曾谋面,却相遇于斯,实在是莫大的缘分,感谢您的到访 ! |