【LeetCode：3153】所有数对中数位差之和（Java）

题目链接

• 3153. 所有数对中数位差之和

题目描述

你有一个数组 nums ，它只包含 正 整数，所有正整数的数位长度都 相同 。

两个整数的 数位差 指的是两个整数 相同 位置上不同数字的数目。

请你返回 nums 中 所有 整数对里，数位差之和。

示例 1：

输入：nums = [13,23,12]

输出：4

解释：
计算过程如下：

• 13 和 23 的数位差为 1 。
• 13 和 12 的数位差为 1 。
• 23 和 12 的数位差为 2 。
  所以所有整数数对的数位差之和为 1 + 1 + 2 = 4 。

示例 2：

输入：nums = [10,10,10,10]

输出：0

解释：
数组中所有整数都相同，所以所有整数数对的数位不同之和为 0 。

提示：

2 <= nums.length <= 10⁵
1 <= nums[i] < 10⁹
nums 中的整数都有相同的数位长度。

求解思路

• 依次计算每一数位上数位不同的次数，例如样例1中，先算个位上不同的次数再算十位上不同的次数。
• cnt[]数组用来记录当前位每个数字出现的次数，例如样例1中，在算个位时，cnt[2] = 1，cnt[3] = 2；在算十位上时，cnt[1] = 2，cnt[2] = 1。
• 完成记录之后，可以计算这一位上数位不同的次数为 $(\sum _{i=0}^9(n-cnt[i])\ast cnt[i])/2$。例如样例1中，在个位数位上，2出现了 cnt[2] 次，那么其他数字就会出现 n − cnt[2] 次，那么数位不同的次数中，包含 2 的情况就有 cnt[2] * (n − cnt[2]) 次。
• 但是这个次数，会在计算其他数字时也被计算一次，所以最后结果要除以 2。（需要注意是最后结果除以2，因为直接除以2的话可能会导致整数除不尽的误差）

实现代码

class Solution {
    public long sumDigitDifferences(int[] nums) {
        long ans = 0;
        int n = nums.length;
        while (nums[0] > 0) {
            int[] cnt = new int[10];
            for (int i = 0; i < n; i ++) {
                cnt[nums[i] % 10] ++;
                nums[i] /= 10;
            }
            for (int i = 0; i < 10; i ++) {
                ans += (long) (n - cnt[i]) * cnt[i];
            }
        }
        return ans / 2;
    }
}