LeetCode 1814. 统计一个数组中好对子的数目（哈希）

Michael阿明

发布于 2021-09-06 10:40:28

2620

发布于 2021-09-06 10:40:28

文章被收录于专栏：Michael阿明学习之路

文章目录

- 1. 题目
- 2. 解题

1. 题目

给你一个数组 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 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

条件等价于 nums[i] - rev(nums[i]) == nums[j] - rev(nums[j])
使用 nums[i] - rev(nums[i]) 作为哈希的 key，计算数量
每个key内的组合都可以， Cn2

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阿明），一起加油、一起学习进步！

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2021/04/04 ，如有侵权请联系 cloudcommunity@tencent.com 删除

https