【Leetcode 1512 】 好数对的数目—— 数组模拟哈希表 与 等差数列求和

给你一个整数数组 nums 。

如果一组数字 (i,j) 满足 nums[i] == nums[j] 且 i < j ，就可以认为这是一组 好数对 。

返回好数对的数目。

示例 1：

输入： nums = [1,2,3,1,1,3]
输出： 4
解释： 有 4 组好数对，分别是 (0,3), (0,4), (3,4), (2,5) ，下标从 0 开始

示例 2：

输入： nums = [1,1,1,1]
输出： 6
解释： 数组中的每组数字都是好数对

示例 3：

输入： nums = [1,2,3]
输出： 0

数组模拟哈希表 与 等差数列求和

// 数组模拟哈希表 与 等差数列求和
function numIdenticalPairs(nums: number[]): number {
  const arr = Array(101).fill(0);
  for (const n of nums) {
    arr[n]++;
  }
  //等差数列求和
  return arr.reduce((pre, cur) =>
    cur === 0 ? pre : (pre += (cur * (cur - 1)) / 2)
  );
}

哈希表

// 哈希表
function numIdenticalPairs2(nums: number[]): number {
  const map = new Map();
  let res = 0;
  for (const n of nums) {
    map.set(n, (map.get(n) || 0) + 1);
    if (map.get(n) > 1) {
      res += map.get(n) - 1;
    }
  }
  return res;
}