class Solution {
public:
int countNicePairs(vector<int>& nums) {
}
};
1814. 统计一个数组中好对子的数目
给你一个数组 nums ，数组中只包含非负整数。定义 rev(x) 的值为将整数 x 各个数字位反转得到的结果。比方说 rev(123) = 321 ， rev(120) = 21 。我们称满足下面条件的下标对 (i, j) 是 好的 ：
0 <= i < j < nums.lengthnums[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 <= 1050 <= nums[i] <= 109原站题解
golang 解法, 执行用时: 88 ms, 内存消耗: 9 MB, 提交时间: 2023-01-17 09:56:27
func countNicePairs(nums []int) (ans int) {
cnt := map[int]int{}
for _, num := range nums {
rev := 0
for x := num; x > 0; x /= 10 {
rev = rev*10 + x%10
}
ans += cnt[num-rev]
cnt[num-rev]++
}
return ans % (1e9 + 7)
}
javascript 解法, 执行用时: 104 ms, 内存消耗: 49.8 MB, 提交时间: 2023-01-17 09:56:09
/**
* @param {number[]} nums
* @return {number}
*/
var countNicePairs = function(nums) {
const MOD = 1000000007;
let res = 0;
const h = new Map();
for (const i of nums) {
let temp = i, j = 0;
while (temp > 0) {
j = j * 10 + temp % 10;
temp = Math.floor(temp / 10);
}
res = (res + (h.get(i - j) || 0)) % MOD;
h.set(i - j, (h.get(i - j) || 0) + 1);
}
return res;
};
java 解法, 执行用时: 29 ms, 内存消耗: 51.7 MB, 提交时间: 2023-01-17 09:52:31
class Solution {
public int countNicePairs(int[] nums) {
final int MOD = 1000000007;
int res = 0;
Map<Integer, Integer> h = new HashMap<Integer, Integer>();
for (int i : nums) {
int temp = i, j = 0;
while (temp > 0) {
j = j * 10 + temp % 10;
temp /= 10;
}
res = (res + h.getOrDefault(i - j, 0)) % MOD;
h.put(i - j, h.getOrDefault(i - j, 0) + 1);
}
return res;
}
}
python3 解法, 执行用时: 284 ms, 内存消耗: 22.7 MB, 提交时间: 2023-01-17 09:51:30
# 哈希表
class Solution:
def countNicePairs(self, nums: List[int]) -> int:
res = 0
cnt = Counter()
for i in nums:
j = int(str(i)[::-1])
res += cnt[i - j]
cnt[i - j] += 1
return res % (10 ** 9 + 7)