中等

问题描述

给你一个数组 nums，数组中只包含非负整数。定义 rev(x) 的值为将整数 x 各个数字位反转得到的结果。比方说 rev(123) = 321，rev(120) = 21。我们称满足下面条件的下标对 (i, j) 是 好的 ：

• 0 <= i < j < nums.length
• nums[i] + rev(nums[j]) == nums[j] + rev(nums[i])

请你返回好下标对的数目。由于结果可能会很大，请将结果对 109 + 7 取余 后返回。

示例 1：

输入：nums = [42,11,1,97]
输出：2
解释：两个坐标对为：
- (0,3)：42 + rev(97) = 42 + 79 = 121, 97 + rev(42) = 97 + 24 = 121 。
- (1,2)：11 + rev(1) = 11 + 1 = 12, 1 + rev(11) = 1 + 11 = 12 。

示例 2：

输入：nums = [13,10,35,24,76]
输出：4

提示：

• 1 <= nums.length <= 105
• 0 <= nums[i] <= 109

哈希表

对于 $nums[i] + rev(nums[j]) == nums[j] + rev(nums[i])$ 可以转换为 $nums[i] - rev(nums[i]) == nums[j] + rev(nums[j])$，因此，可以遍历数组，计算 $nums[i] - rev(nums[i])$ 的结果，统计相同结果的个数存到哈希表中，后续遇到相同的结果时，将已存在哈希表中的结果个数加到答案中即可。

/**
 * @param {number[]} nums
 * @return {number}
 */
var countNicePairs = function (nums) {
  const MOD = Math.pow(10, 9) + 7
  const map = new Map()
  const rev = (num) => Number(num.toString().split('').reverse().join(''))
  let ans = 0
  for (const num of nums) {
    const res = num - rev(num)
    if (!map.has(res)) {
      map.set(res, 1)
    } else {
      const cnt = map.get(res)
      ans = (ans + cnt) % MOD
      map.set(res, cnt + 1)
    }
  }
  return ans
}

• 时间复杂度：$O(n \times \log{C})$
• 空间复杂度：$O(n)$