b_lc_统计一个数组中好对子的数目(数学思路)

我们称满足下面条件的下标对 (i, j) 是 好的 :

  • 0 <= i < j < nums.length
  • nums[i] + rev(nums[j]) == nums[j] + rev(nums[i])
    请你返回好下标对的数目 10^9 + 7 取余结果。
输入: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 。

思路:如果两个数a、b能组对,a-rev(a) = b-rev(b),因为根据组合公式(C_n^2):n*(n-1)/2

typedef long long ll;
const int mod=1e9+7;
class Solution {
public:
    int rev(int x) {
        int n=0;
        while (x) {
            n = n*10 + x % 10;
            x /= 10;
        }
        return n;
    }
    int countNicePairs(vector<int>& A) {
        int n=A.size();
        unordered_map<ll, ll> mp;
        for (int x : A){
            mp[x-rev(x)]++;
        }
        ll ans=0;
        for (auto& [k,v] : mp) {
            if (v < 2) continue;
            ans=(ans + v * (v-1) / 2l % mod) % mod;
        }
        return ans % mod;
    }
};
原文地址:https://www.cnblogs.com/wdt1/p/14615388.html