**454. 4Sum II 四数之和

1. 原始题目

给定四个包含整数的数组列表 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

2. 思路

注意≤ N ≤ 500提示我们O（n^2）是可以接受的。

利用查找表将4次方复杂度转为平方复杂度。具体方法:将C+D结果都存在查找表字典中。然后根据再遍历A+B寻找是否有满足条件的元素。

3. 解题

 1 class Solution:
 2     def fourSumCount(self, A, B, C, D) -> int:
 3         s = dict()
 4         res = 0
 5         for i in range(len(C)):
 6             for j in range(len(D)):
 7                 if C[i]+D[j] not in s:
 8                     s[C[i]+D[j]] = 1
 9                 else:
10                     s[C[i]+D[j]] += 1
11 
12         for i in range(len(A)):
13             for j in range(len(B)):
14                 if -(A[i]+B[j]) in s:
15                     res+=s[-(A[i]+B[j])]
16         return res

leetcode上的更简洁代码：

1 def fourSumCount(self, A, B, C, D):
2     AB = collections.Counter(a+b for a in A for b in B)
3     return sum(AB[-c-d] for c in C for d in D)

 

posted @ 2019-05-05 21:27  三年一梦  阅读(146)  评论(0)    收藏  举报