统计一个数组中好对子的数目
难度:中等
给你一个数组 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
哈希表
思路:
首先题目给出一个下标从
0
0
0 开始长度为
n
n
n 的非负整数数组
nums
\textit{nums}
nums,并给出「好下标对」的定义——对于某一个下标对
(
i
,
j
)
(i,j)
(i,j),
0
≤
i
<
j
<
n
0 \le i < j < n
0≤i<j<n,若满足:
nums [ i ] + rev ( nums [ j ] ) = nums [ j ] + rev ( nums [ i ] ) (1) \textit{nums}[i] + \textit{rev}(\textit{nums}[j]) = \textit{nums}[j] + \textit{rev}(\textit{nums}[i]) \tag{1} nums[i]+rev(nums[j])=nums[j]+rev(nums[i])(1)
则该下标对为「好下标对」。现在我们设: f ( i ) = nums [ i ] − rev ( nums [ i ] ) f(i) = \textit{nums}[i] - \textit{rev}(\textit{nums}[i]) f(i)=nums[i]−rev(nums[i]),则表达式 ( 1 ) (1) (1) 可以等价于:
f ( i ) = f ( j ) (2) f(i) = f(j) \tag{2} f(i)=f(j)(2)
那么我们从左到右遍历数组 nums \textit{nums} nums,并在遍历的过程用「哈希表」统计每一个位置 i i i, 0 ≤ i < n 0 \le i < n 0≤i<n 的 f ( i ) f(i) f(i) 的个数,则对于位置 j j j, 0 ≤ j < n 0 \le j < n 0≤j<n,以 j j j 结尾的「好下标对」的个数为此时「哈希表」中 f ( j ) f(j) f(j) 的数目。那么我们只需要在遍历时同时统计以每一个位置为结尾的「好下标对」数目即可。
复杂度分析:
- 时间复杂度: O ( n × log C ) O(n \times \log C) O(n×logC),其中 n n n 为数组 nums \textit{nums} nums 的长度, C C C 为数组 nums \textit{nums} nums 中的数字范围。其中 O ( log C ) O(\log C) O(logC) 为反转数位的时间复杂度。
- 空间复杂度: O ( n ) O(n) O(n),其中 n n n 为数组 nums \textit{nums} nums 的长度,主要为哈希表的空间开销。
class Solution:
def countNicePairs(self, nums: List[int]) -> int:
diff_count, res = dict(), 0
for i in nums:
diff = i - int(str(i)[::-1])
if diff_count.get(diff):
res += diff_count[diff]
diff_count[diff] += 1
else:
diff_count[diff] = 1
return res % (10 ** 9 + 7)
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/count-nice-pairs-in-an-array