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[j]) == nums[j] + rev(nums[i])
可以转变为nums[i]-rev(nums[i])==nums[j]-rev(nums[j])
进而题目可以转换为求和问题：
对于每个i位置的nums[i]-rev(nums[i])，数组有多少个与之相等的j位置，且j>i。
从而可以先计算出所有的nums[i]-rev(nums[i])，存储在nrevs数组中，
同时用一个Map进行存储，然后遍历nrevs数组，累加结果。

代码

class Solution {
    //nums[i]-rev(nums[i])==nums[j]-rev(nums[j])
    public int countNicePairs(int[] nums) {
        Map<Integer,Integer> map=new HashMap<>();
        int n=nums.length;
        int[] nrevs=new int[n];
        for(int i=0;i<n;i++){
            int rev=revs(nums[i]);
            nrevs[i]=nums[i]-rev;
            map.put(nrevs[i],map.getOrDefault(nrevs[i],0)+1);
        }
        long ans=0l;
        for(int i=0;i<n;i++){
            int count=map.get(nrevs[i]);
            ans+=(count-1);
            map.put(nrevs[i],count-1);
        }
        ans=ans%1000000007;
        return (int)ans;
    }
    int revs(int item){
        int ans=0;
        while(item>0){
            ans=ans*10+item%10;
            item=item/10;
        }
        return ans;
    }
}