leetcode 做题记录 - 454.四数相加 II

原题

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;
};

声明

题目来源于力扣，侵删。

以上纯为个人理解写下，能力有限，如有错误，欢迎指出，让我学习。感谢！