-
Notifications
You must be signed in to change notification settings - Fork 3
Expand file tree
/
Copy path1814.cpp
More file actions
28 lines (28 loc) · 692 Bytes
/
1814.cpp
File metadata and controls
28 lines (28 loc) · 692 Bytes
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
class Solution {
public:
int rev(int num) {
int res = 0;
while (num) {
res *= 10;
res += num % 10;
num /= 10;
}
return res;
}
int countNicePairs(vector<int>& nums) {
int n = nums.size();
unordered_map<int, int> counts;
for (int i = 0; i < n; ++i) {
int computed = nums[i] - rev(nums[i]);
counts[computed] += 1;
}
long long res = 0;
long long mod = 1e9 + 7;
for (auto& p : counts) {
long long count = p.second;
res += (count * (count - 1)) >> 1;
res %= mod;
}
return res;
}
};