1814. 统计一个数组中好对子的数目Java
文章发布较早,内容可能过时,阅读注意甄别。
# 题目
给你一个数组 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
# 思路
哈希表
# 解法
class Solution {
final int mod = (int)1e9 + 7;
public int countNicePairs(int[] nums) {
Map<Integer, Integer> map = new HashMap<>();
int n = nums.length;
long cnt = 0L;
for (int i : nums) {
int ri = rev(i);
map.put(i - ri, map.getOrDefault(i - ri, 0) + 1);
}
for (int i = 0; i < n; i++) {
int u = nums[i] - rev(nums[i]);
map.put(u, map.get(u) - 1);
cnt += map.getOrDefault(u, 0);
cnt %= mod;
}
return (int)cnt;
}
int rev(int u) {
int res = 0;
while(u != 0) {
res = res * 10 + u % 10;
u /= 10;
}
return res;
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
# 总结
- 分析出几种情况,然后分别对各个情况实现