Count Nice Pairs in an Array LeetCode solution in Python

1
class Solution:
2
    def countNicePairs(self, nums: List[int]) -> int:
3
        res = 0
4
        count = {}
5
        mod = 10**9 + 7
6

7
        for n in nums:
8
            rev = int(str(n)[::-1])
9
            cur = count.get(n - rev, 0)
10
            res += cur
11
            count[n - rev] = 1 + cur
12

13
        return res % mod