【LeetCode每日一题:1814. 统计一个数组中好对子的数目~~~数字反转+HashMap计数+结果取余】

题目描述

给你一个数组 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])
请你返回好下标对的数目。由于结果可能会很大,请将结果对 109 + 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 <= 105
0 <= nums[i] <= 109

求解思路

  1. 该题目仔细想想可以想到通过HashMap计数来做。但是怎么做需要考虑一下。
  2. 解题的突破口就在题目给定的提示中,先通过循环遍历每一个位置的结果,最后计数即可。
    在这里插入图片描述

实现代码

class Solution {
    
    
    public int countNicePairs(int[] nums) {
    
    
        final int MOD = 1000000007;
        int res = 0;
        Map<Integer, Integer> h = new HashMap<Integer, Integer>();
        for (int i : nums) {
    
    
            int temp = i, j = 0;
            while (temp > 0) {
    
    
                j = j * 10 + temp % 10;
                temp /= 10;
            }
            res = (res + h.getOrDefault(i - j, 0)) % MOD;
            h.put(i - j, h.getOrDefault(i - j, 0) + 1);
        }
        return res;
    }
}

运行结果

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/Coder_ljw/article/details/128714151