【Leetcode 2367 】 等差三元组的数目 —— 哈希表与三指针

给你一个下标从 0 开始、严格递增 的整数数组 nums 和一个正整数 diff 。如果满足下述全部条件，则三元组 (i, j, k) 就是一个 等差三元组 ：

• i < j < k ，
• nums[j] - nums[i] == diff 且
• nums[k] - nums[j] == diff

返回不同 等差三元组 的数目 。

示例 1：

输入： nums = [0,1,4,6,7,10], diff = 3
输出： 2
解释：
(1, 2, 4) 是等差三元组：7 - 4 == 3 且 4 - 1 == 3 。
(2, 4, 5) 是等差三元组：10 - 7 == 3 且 7 - 4 == 3 。

示例 2：

输入： nums = [4,5,6,7,8,9], diff = 2
输出： 2
解释：
(0, 2, 4) 是等差三元组：8 - 6 == 2 且 6 - 4 == 2 。
(1, 3, 5) 是等差三元组：9 - 7 == 2 且 7 - 5 == 2 。

提示：

• 3 <= nums.length <= 200
• 0 <= nums[i] <= 200
• 1 <= diff <= 50
• nums 严格 递增

哈希表

/*
https://leetcode.cn/u/cshappyeveryday/
执行用时：59 ms, 在所有 Typescript 提交中击败了 100.00 %的用户
内存消耗：51.72 MB, 在所有 Typescript 提交中击败了 100.00 %的用户
时间复杂度：O(n) , 空间复杂度: O(n)
2024年9月2日 
*/
function arithmeticTriplets(nums: number[], diff: number): number {
  const numSet = new Set(nums);
  return nums.reduce(
    (pre, cur) =>
      numSet.has(diff + cur) && numSet.has(2 * diff + cur) ? pre + 1 : pre,
    0
  );
}

三指针

/*
https://leetcode.cn/u/cshappyeveryday/
执行用时：58 ms, 在所有 Typescript 提交中击败了 100.00 %的用户
内存消耗：51.8 MB, 在所有 Typescript 提交中击败了 100.00 %的用户
时间复杂度：O(n) , 空间复杂度: O(1)
2024年9月2日 
*/
function arithmeticTriplets(nums: number[], diff: number): number {
  const n = nums.length;
  let ans = 0;
  for (let i = 0, j = 1, k = 2; i < n - 2 && j < n - 1 && k < n; i++) {
    //1、比较 j 与 i 、nums[j] 与 nums[i]

    //防止出现 [1,4,8,20]  diff = 1,在循环时，将直接会continue，i++，而j没有移动，此时j将===i
    //为j重新赋值
    j = Math.max(j, i + 1);

    //将中间指针移动到 nums[j]-nums[i] >= diff 的位置上
    while (j < n - 1 && nums[j] - nums[i] < diff) {
      j++;
    }
    // nums[j]-nums[i] > diff ,则该左指针没有符合结果的值，跳过
    if (j >= n - 1 || nums[j] - nums[i] > diff) {
      continue;
    }

    //防止出现 [1,4,8,20]  diff = 1,循环时，将不会执行k++，此时k有可能在循环后 < j
    //为k重新赋值
    k = Math.max(k, j + 1);

    //上面比较完，则比较 k 与 j，nums[k] 与 nums[j]
    while (k < n && nums[k] - nums[j] < diff) {
      k++;
    }
    if (k < n && nums[k] - nums[j] === diff) {
      ans++;
    }
  }
  return ans;
}