统计一个数组中好对子的数目---LeetCode练习

这次leetcode的周赛好多坑啊，还是我太菜了

题目描述

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

思路

假如一个数x，它反转后是y

假如一个数a，它反转后是b

如果x-y=a-b，则这两个数满足条件

因为 x+b=a+y

当有两个数的原数和反转后的数的差值相等，那么坐标数为1

当有三个数的原数和反转后的数的差值相等，那么坐标数为1+2

当有四个数的原数和反转后的数的差值相等，那么坐标数为1+2+3

我这里用的字典，先遍历一次nums数组，找到每个数反转后与原数的差值a，然后存入字典，如果字典里没有那就dt[a]=1，如果字典里有那就dt[a]+=1，最后字典里是各个差值的数量。

算组合的总数就好了。

代码

class Solution:
    def countNicePairs(self, nums: List[int]) -> int:
        numbers=dict()
        for i in range(len(nums)):
            s1=str(nums[i])
            s2=0
            for j in range(len(s1)):
                s2+=int(s1[j])*10**j
            a=nums[i]-s2
            if a in numbers:
                numbers[a]+=1
            else:
                numbers[a]=1
        kinds=0
        for key in numbers.keys():
            a=numbers[key]
            for i in range(1,a):
                kinds+=i
        return(kinds%(10**9+7))

编码不易，有帮助的话点个赞支持一下吧