给你一个数组 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])
请你返回好下标对的数目。由于结果可能会很大,请将结果对 10^9 + 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 <= 10^5
0 <= nums[i] <= 10^9
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/count-nice-pairs-in-an-array 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
nums[i] - rev(nums[i]) == nums[j] - rev(nums[j])
nums[i] - rev(nums[i])
作为哈希的 key, 计算数量class Solution {
public:
int countNicePairs(vector<int>& nums) {
long long n = nums.size(), ans = 0, mod = 1e9+7;;
map<int, long long> m;
for(int i = 0; i < n; i++) {
int n = nums[i], rev = 0;
while(n)
{
rev = rev*10 + n%10;
n /= 10;
}
m[nums[i]-rev]++;
}
for(auto it = m.begin(); it != m.end(); ++it)
{
ans = (ans + it->second*(it->second-1)/2)%mod;
}
return ans;
}
};
120 ms 44.2 MB C++
我的CSDN博客地址 https://michael.blog.csdn.net/
长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!