开启掘金成长之旅！这是我参与「掘金日新计划 · 2 月更文挑战」的第 22 天，点击查看活动详情

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

给你一个数组 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[i]) 的值，而不必去计算不同的数对，我们获取每一个差值的结果后，保存该结果的个数，然后就可以得到不同数对的个数。

对于同一个结果的值来说，我们可以计算得到的数对的个数，若有 n 个值的计算结果相同，则不同的数对个数为 (n - 1) * n / 2，我们累加所有的结果，就可以得到最终答案，在累加的同时，需要注意对所得结果进行取余判断。

题解

class Solution {
    public int countNicePairs(int[] nums) {
        long cnt = 0;
        final int MOD = (int) 1e9 + 7;
        int n = nums.length;
        Map<Integer, Long> map = new HashMap<>();
        for(int i = 0; i < n; i++) {
            int ans = nums[i] - rev(nums[i]);
            map.put(ans, map.getOrDefault(ans, 0L) + 1);
        }
        for(int key: map.keySet()) {
            long val = map.get(key);
            cnt += val * (val - 1) / 2;
            cnt %= MOD;
        }
        return (int) cnt;
    }

    private int rev(int x) {
        int res = 0;
        while(x > 0) {
            res = res * 10 + x % 10;
            x /= 10;
        }
        return res;
    }
}

如果你有其他的思路或者更好的解法，亦或者你发现了文章出现了错误或有不足，欢迎在评论区和我交流，我看到了一定会回复。

写文章不易，如果你觉得文章对你有帮助，麻烦点一下点赞、收藏，你的支持是我写文章的最大动力！