一. 题目描述
给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l)
,使得 A[i] + B[j] + C[k] + D[l] = 0
。
为了使问题简单化,所有的 A, B, C, D 具有相同的长度 N,且 0 ≤ N ≤ 500 。所有整数的范围在 -228 到 228 - 1 之间,最终结果不会超过 231 - 1 。
例如:
输入: A = [ 1, 2] B = [-2,-1] C = [-1, 2] D = [ 0, 2] 输出: 2 解释: 两个元组如下: 1. (0, 0, 0, 1) -> A[0] + B[0] + C[0] + D[1] = 1 + (-2) + (-1) + 2 = 0 2. (1, 1, 0, 0) -> A[1] + B[1] + C[0] + D[0] = 2 + (-1) + (-1) + 0 = 0
二. 解题思路
这里直接放官方的解题思路,不过这些情况我也都考虑到了,并同样采用了第三种方式,可惜还是没有官方的方法简洁,有兴趣的可以直接看官方的代码。
- 暴力解法,时间复杂度 O(n^4)
- 将 D 中的元素放入查找表,时间复杂度 O(n^3)
- 将 A + B 的每一种可能放入查找表,然后 两重循环对在查找表中寻找是否存在 -(C[i] + D[i]),时间复杂度为 O(n^2)。两数相加的和作为键值。
我采用第三种方法实现
三. 代码
class Solution(object):
def fourSumCount(self, A, B, C, D):
"""
:type A: List[int]
:type B: List[int]
:type C: List[int]
:type D: List[int]
:rtype: int
"""
dic_1 = {}
dic_2 = {}
for idx, num_1 in enumerate(A):
for idy, num_2 in enumerate(B):
result = num_1 + num_2
if result in dic_1:
_list = dic_1.get(result)
_list.append((idx, idy))
dic_1[result] = _list
else:
_list = [(idx, idy)]
dic_1[result] = _list
for idx, num_1 in enumerate(C):
for idy, num_2 in enumerate(D):
result = num_1 + num_2
if result in dic_2:
_list = dic_2.get(result)
_list.append((idx, idy))
dic_2[result] = _list
else:
_list = [(idx, idy)]
dic_2[result] = _list
result = 0
for number_1 in dic_1.keys():
target = 0 - number_1
if target in dic_2:
list_1 = dic_1.get(number_1)
list_2 = dic_2.get(target)
result += len(list_1) * len(list_2)
return result
四. 参考
1. https://leetcode-cn.com/explore/orignial/card/all-about-lockup-table/239/learn-to-use-keys/998/