思路
因为所有整数都有相同的数位长度,所以先算出数位长度,然后依次统计个位、十位、百位…的数位不同之和,最后除以2
解题过程
例如:13、12的数位不同之和,可以简化为个位2、3和十位1、1的数位不同之和,即答案为1。需要注意(2,3)会被计算两次,所以最终答案需要除以2
Code
class Solution {
public long sumDigitDifferences(int[] nums) {
int len=nums.length;
int n=nums[0];
int nLen=0; //数位长度
while(n>0){
n/=10;
nLen++;
}
long arr[][]=new long[nLen][10];
for(int i=0;i<nLen;i++){
for(int j=0;j<len;j++){
arr[i][nums[j]%10]++;
nums[j]/=10;
}
}
long ans=0;
for(int i=0;i<nLen;i++){
for(int j=0;j<10;j++){
ans+=arr[i][j]*(len-arr[i][j]);
}
}
return ans/2;
}
}
作者:菜卷
链接:https://leetcode.cn/problems/sum-of-digit-differences-of-all-pairs/solutions/2897598/suo-you-shu-dui-zhong-shu-wei-bu-tong-zh-cbc4/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。