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

1、题干

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

2、思路

题中的条件转换为 nums[i] - rev(nums[i]) == nums[j] - rev(nums[j])，转换后简单很多，遍历所有数字计算 nums[i] - rev(nums[i]) 并使用哈希表计数

3、代码

function countNicePairs(nums: number[]): number {
    const M = 1e9 + 7, map = new Map();

    let ans = 0;
    for (const n of nums) {
        let cn = n, rn = 0;
        while (cn) {
            rn = 10 * rn + (cn % 10);
            cn = (cn / 10) >> 0;
        }

        const c = map.get(n - rn) || 0;
        ans = (ans + c) % M;
        map.set(n - rn, c + 1);
    }

    return ans;
};

可以使用字符串反转数字，代码更简短，但消耗的时间和内存偏多（时间：204 ms，内存：63.5 MB）。

function countNicePairs(nums: number[]): number {
    const M = 1e9 + 7, map = new Map();

    for (const n of nums) {
        const rn = +[...`${n}`].reverse().join('');
        map.set(n - rn, (map.get(n - rn) || 0) + 1);
    }

    return [...map.values()].reduce((a, c) => (a + (c * c - c) / 2) % M, 0);
};

4、复杂度

• 时间复杂度：$O(n*logm)$，其中 $n$ 为数组长度，$m$ 为数字值
• 空间复杂度：$O(n)$

5、执行结果