第3节 位数差
给一个数组{a},定义 h(a,b)为在十进制下 a + b 与 a 的位数差,求 ,0的位数为1。
输入描述:
第一行读入一个正整数 n (1 <= n <= 105)。
第二行读入 n 个非负整数,第 i 个表示a[i] (0 <= a[i] <= 108)。
输出描述:
一行表示答案。
示例1
输入
10
0 1 2 3 4 5 6 7 8 9
输出
20
把数组升序排列,那么当前数只会加上比它小的数,要么进1位,要么不进位,二分找到第一个不进位的位置,从而统计a[i]+a[j]的位数之和
减去(n-i)*a[i]的个数和,就是ans
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e5+5;
ll a[N];
ll bit(ll x)//计算位数
{
if(x==0) return 1;
ll cot=0;
while(x)
{
x/=10;
cot++;
}
return cot;
}
bool cmp(ll x,ll y) {
return x>y;
}
ll ans=0;
int main()
{
// freopen("test.txt", "r", stdin);
int n;
cin>>n;
for(int i=1;i<=n;i++)
{
scanf("%lld",&a[i]);
ans-=(n-i)*bit(a[i]);//第i个数后面有n-i个数
}
sort(a+1,a+1+n,cmp);//任意两个组合数之和与顺序无关
for(int i=1;i<=n-1;i++)
{
ll l=i+1,r=n+1;
ll now=bit(a[i]);
while(l<r)//左闭右开,当l==r时跳出循环,此时的l或r都为右段的第一个数的位置,如1111111000000,l=r=第一个0的位置
//对应题目中第一个 和 无进位的a[j]
{
ll mid=(l+r)>>1;
if(bit(a[i]+a[mid])>now) l=mid+1;
else r=mid;
}
//cout<<l-1-i<<","<<n-l+1<<endl;
ans+=(n-l+1)*now+(l-1-i)*(now+1);
//1的个数为l-i+1, 0的个数为n-l+1
//cout<<ans<<endl;
}
cout<<ans<<endl;
return 0;
}