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

解题思路

这里借鉴代码随想录的方法：

将四个数组两两分开，整体上可以看为是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