1814 统计一个数组中好对子的数目（字典计数、组合）

最新推荐文章于 2024-02-23 21:35:20 发布

原创最新推荐文章于 2024-02-23 21:35:20 发布 · 319 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

力扣

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-nice-pairs-in-an-array

2. 思路分析：

分析题目可以知道最容易想到的是暴力枚举每一组的数字然后计算是否满足题目中好下标对的要求，但是这样肯定会超时，一般题目给出的式子我们想能不能简化一下呢？其实是可以的，通过对等式进行移项将所有与当前数字有关的操作放在一起，可以发现如果两个数字满足nums[i] - rev(nums[i]) == nums[j] - rev(nums[j])他们就是好下标对，所以我们可以遍历nums列表，找出nums[i] - rev(nums[i])相等的数目，具体的做法是使用字典存储起来，这样所有差值相等的都可以存储到一起，最后遍历字典中存储的值，因为是组合问题，所有n个元素的组合方式为：n * (n - 1) // 2，举一个简单的例子就好理解了，所以在遇到与等式相关的问题，首先想到的是能否对等式进行化简呢。

3. 代码如下：

import collections
from collections import Counter


class Solution:
    def countNicePairs(self, nums):
        dic = collections.defaultdict(int)
        # 字典记录差值相等的数目, [::-1]表示取出序列的逆序
        for num in nums:
            dic[num - int(str(num)[::-1])] += 1
        res = 0
        for value in dic.values():
            # 求解组合的数目
            res += value * (value - 1) // 2
        return res % (10 ** 9 + 7)