1814. 统计一个数组中好对子的数目

最新推荐文章于 2025-08-31 20:23:04 发布

原创最新推荐文章于 2025-08-31 20:23:04 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #哈希算法

Leetcode 专栏收录该内容

144 篇文章

订阅专栏

文章介绍了一种使用哈希表解决编程题目的方法，通过建立哈希表存储数组元素的反转值的出现次数，遍历数组计算每个元素与其反转值的差值，从而找到满足条件的数对。时间复杂度为O(nlogm)，空间复杂度为O(n)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

解法一：哈希表应用

首先对于题目的式子 $i$ 和 $j$ 都出现在了两边，我们将其统一移动在一方

nums[i] + rev(nums[j]) = nums[j] + rev(nums[i]) -> nums[i] - rev(nunm[i] = nums[j] - rev(nums[j])

那么问题就转化为：求出 $(i, j)$ 对，使得 $n u m s [i] - re v (n u nm [i] = n u m s [j] - re v (n u m s [j])$ , 很明显我们可以用哈希表来求解。用哈希表来保存某个数出现的次数，我们从前往后遍历数组中所有数，对于当前数x来说，我们求得 $x - re v (x)$ ，由于哈希表里面保存的都是以前某个nums[i]+rev(nums[i])出现的次数，那么我们直接累加以前 $x - re v (x)$ 出现次数即可。

时间复杂度： $O (n l o g m)$ , n为数组长度，m为数组中最大值
空间复杂度： $O (n)$
相似题目：和为k的子数组 | 连续的子数组和统计中位数为 K 的子数组

class Solution {
    int mod = (int)1e9 + 7;
    public int countNicePairs(int[] nums) {
        Map<Integer, Integer> mp = new HashMap<>();
        int ans = 0;
        for (int x : nums) {
            int rev = 0;
            for (int t = x; t != 0; t /= 10) rev = rev * 10 + t % 10;
            ans = (ans + mp.getOrDefault(x - rev, 0)) % mod; 
            mp.put(x - rev, mp.getOrDefault(x - rev, 0) + 1);
        }
        return ans;
    }
}

class Solution {
public:
    int countNicePairs(vector<int>& nums) {
        int mod = 1e9 + 7;
        unordered_map<int, int> mp;
        int ans = 0;
        for (int x : nums) {
            int rev = 0;
            for (int t = x; t != 0; t /= 10) rev = rev * 10 + t % 10;
            ans = (ans + mp[x - rev]) % mod; 
            mp[x - rev]++;
        }
        return ans;
    }
};