【Leetcode 1995 】 统计特殊四元组 —— 最全的注解

给你一个 下标从 0 开始 的整数数组 nums ，返回满足下述条件的 不同 四元组 (a, b, c, d) 的 数目 ：

• nums[a] + nums[b] + nums[c] == nums[d] ，且
• a < b < c < d

示例 1：

输入： nums = [1,2,3,6]
输出： 1
解释： 满足要求的唯一一个四元组是 (0, 1, 2, 3) 因为 1 + 2 + 3 == 6 。

示例 2：

输入： nums = [3,3,6,4,5]
输出： 0
解释： [3,3,6,4,5] 中不存在满足要求的四元组。

示例 3：

输入： nums = [1,1,1,3,5]
输出： 4
解释： 满足要求的 4 个四元组如下：
- (0, 1, 2, 3): 1 + 1 + 1 == 3
- (0, 1, 3, 4): 1 + 1 + 3 == 5
- (0, 2, 3, 4): 1 + 1 + 3 == 5
- (1, 2, 3, 4): 1 + 1 + 3 == 5

最全的注解

function countQuadruplets(nums: number[]): number {
  const cnt = new Map<number, number>();
  let ans = 0;
  const len = nums.length;
  // a,b,c,d
  //由于 nums[a] + nums[b] + nums[c] = nums[d]
  //所以 nums[a] + nums[b] = nums[d] - nums[c]
  for (let b = len - 3; b >= 1; b--) {
    for (let a = b - 1; a >= 0; a--) {
      const sum = nums[b] + nums[a];
      cnt.set(sum, (cnt.get(sum) || 0) + 1);
    }
    // 找出 d，其索引最低要比b大2，因为 b < c < d
    for (let d = b + 2, c = b + 1; d < len; d++) {
      // b + 1 代替为c
      // 之所以 c 被固定为 b+1，而不是和d一样向后依次枚举
      //是因为 随着b从后向前枚举时，c同时也从后向前枚举了，
      //而在每次b移动时，d都要把后面的枚举一遍，相当于也和不同的c进行了匹配
      const diff = nums[d] - nums[c];
      if (cnt.has(diff)) {
        ans += cnt.get(diff)!;
      }
    }
  }
  return ans;
}