原题
454. 四数相加 II
给定四个包含整数的数组列表 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
初始解题思路
个人解题时毫无任何新奇的思路,脑子里只能想到4层循环嵌套,但是明显是不对的,在提交以后面对长度特别长且复杂的测试时直接超时。思考的时候有往树相关方向去想,但是感觉明显不对,遂放弃。
直接提交了4层嵌套解法,失败
查看官方题解
题解思路确实厉害:
4层嵌套数据分为两两计算,第一层 A、B 数组两层循环计算出两两配对的值,将其存储在 HashMap 中,计算出的值作为 key,value 为该值计算出现的次数。
即:题目例子中举例
伪代码:
for (a : A)
for (b : B)
map.set(a + b, map.get(a + b) + 1)
1 + (-2) = -1 key = -1 value = 1
1 + (-1) = 0 key = 0 value = 1
2 + (-2) = 0 key = 0 value = 1 + 1
2 + (-1) = 1 key = 1 value = 1
如上,获取到一个 HashMap 结果集,HashMap 中的插入查询时间复杂度为 O(1),循环的时间复杂度为 O(n^2),最终为 O(1) + O(n^2) = O(n^2)。
然后再循环计算,判断 C、D 中的值两项相加的相反值是否存在于 HashMap 中,累加匹配上的 value 值即可。
伪代码:
for (c : C)
for (d : D)
if (map.has(-(c + d)))
count += map.get(-(c + d))
至此,该题解完,HashMap 中的查询复杂度为 O(1),循环时间复杂度为 O(n^2),最终为 O(1) + O(n^2) = O(n^2)。。
总结
总结下,上面解答的时间复杂度为 O(n^2) + O(n^2) = O(n^2),空间复杂度为 HashMap 使用的空间,最坏的情况下 每一项 a + b 值都不相等,即为 O(n^2)。
JavaScript 解法
/**
* @param {number[]} A
* @param {number[]} B
* @param {number[]} C
* @param {number[]} D
* @return {number}
*/
var fourSumCount = function(A, B, C, D) {
let count = 0;
let map = new Map();
for (const a of A) {
for (const b of B) {
map.set(a + b, (map.get(a + b) || 0) + 1);
}
}
for (const c of C) {
for (const d of D) {
if (map.has(-(c + d))) {
count += map.get(-(c + d));
}
}
}
return count;
};
声明
题目来源于力扣,侵删。
以上纯为个人理解写下,能力有限,如有错误,欢迎指出,让我学习。感谢!