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
解释:
两个元组如下:
- (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
- (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
解题思路
这里借鉴代码随想录的方法:
将四个数组两两分开,整体上可以看为是a+b=0,其中a=nums1+nums2,b=nums3+nums4(这里的+理解为两个数组中每个元素依次相加的和),拿a举个例子:
数组 | 值 |
---|---|
nums1 | 1,2 |
nums2 | -2,-1 |
这里相当于nums1[0]+nums2[0](1-2)、nums1[0]+nums2[1](1-1)、nums1[1]+nums2[0](2-2)、nums1[1]+nums2[1](2-1),同理b也是一样的。
将得到的值作为key,以及key出现的次数作为value构建a,b字典。因为a+b=0,所以只需要遍历a的同时判断0-a是否在b中,如果存在,则此时满足题目条件的组合个数为a此时对应键值的个数×b此时对应键值的个数,最后把所有结果累加起来。
如何理解“组合个数为a此时对应键值的个数×b此时对应键值的个数”
举个例子:
a | b |
---|---|
nums1[0]+nums2[0]=2 | nums3[0]+nums4[0]=-2 |
nums1[1]+nums2[0]=2 | nums3[0]+nums4[1]=-2 |
nums1[1]+nums2[2]=2 | |
nums1[2]+nums2[1]=2 |
当nums1[0]+nums2[0]=2时,可与之相加为0的组合有nums3[0]+nums4[0]=-2,nums3[0]+nums4[1]=-2 ,现在在a中有4个2,b中有2个-2,所以满足题目条件的总个数为4×2=8.
补充知识
字典的每个键值 key:value 对用冒号 : 分割,每个键值对之间用逗号 , 分割,整个字典包括在花括号 {} 中 ,格式如下所示:
d = {key1 : value1, key2 : value2 }
常用方法总结:
dict.get(key, default=None) 返回指定键的值,如果值不在字典中返回default值
dict.keys() 以列表返回一个字典所有的键
dict.values() 以列表返回字典中的所有值
分析
详细思路
构建a部分:
ans1={}
for i in nums1:
for j in nums2:
if i+j in ans1.keys():
ans1[i+j] += 1
else:
ans1[i+j] = 1
构建b部分:
ans2={}
for n in nums3:
for k in nums4:
if n+k in ans2.keys():
ans2[n+k] += 1
else:
ans2[n+k] = 1
结果累加:
result=0
for a in ans1.keys():
if -a in ans2.keys():
result+=ans1[a]*ans2[-a]
完整代码:
ans1={}
for i in nums1:
for j in nums2:
if i+j in ans1.keys():
ans1[i+j] += 1
else:
ans1[i+j] = 1
ans2={}
for n in nums3:
for k in nums4:
if n+k in ans2.keys():
ans2[n+k] += 1
else:
ans2[n+k] = 1
result=0
for a in ans1.keys():
if -a in ans2.keys():
result+=ans1[a]*ans2[-a]
return result