算法刷题打卡第68天：统计一个数组中好对子的数目

统计一个数组中好对子的数目

难度：中等

给你一个数组 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

哈希表

思路：
首先题目给出一个下标从 $0$ 开始长度为 $n$ 的非负整数数组 $\text{nums}$，并给出「好下标对」的定义——对于某一个下标对 $(i,j)$，$0\le i<j<n$，若满足：

$$\text{nums}[i]+\text{rev}(\text{nums}[j])=\text{nums}[j]+\text{rev}(\text{nums}[i])\text{\hspace{1em}(1)}$$

则该下标对为「好下标对」。现在我们设：$f(i)=\text{nums}[i]-\text{rev}(\text{nums}[i])$，则表达式 $(1)$ 可以等价于：

$$f(i)=f(j)\text{\hspace{1em}(2)}$$

那么我们从左到右遍历数组 $\text{nums}$，并在遍历的过程用「哈希表」统计每一个位置 $i$，$0\le i<n$ 的 $f(i)$ 的个数，则对于位置 $j$，$0\le j<n$，以 $j$ 结尾的「好下标对」的个数为此时「哈希表」中 $f(j)$ 的数目。那么我们只需要在遍历时同时统计以每一个位置为结尾的「好下标对」数目即可。

复杂度分析：

• 时间复杂度： $O(n\times \log C)$，其中 $n$ 为数组 $\text{nums}$ 的长度，$C$ 为数组 $\text{nums}$ 中的数字范围。其中 $O(\log C)$ 为反转数位的时间复杂度。
• 空间复杂度： $O(n)$，其中 $n$ 为数组 $\text{nums}$ 的长度，主要为哈希表的空间开销。

class Solution:
    def countNicePairs(self, nums: List[int]) -> int:
        diff_count, res = dict(), 0
        for i in nums:
            diff = i - int(str(i)[::-1])
            if diff_count.get(diff):
                res += diff_count[diff]
                diff_count[diff] += 1
            else:
                diff_count[diff] = 1
        return res % (10 ** 9 + 7)

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/count-nice-pairs-in-an-array