牛客算法竞赛入门课第三节习题---位数差

二分+分治

题意:
定义一个函数h(a,b)，h(a,b)是a+b与b的位数差
求数组a中，求
思路：
我们定义一个函数solve(L,R)，求解区间[L,R)(左闭右开)中的h(a,b)个答案，
那么solve(L,R) 的值就等于
solve(L,mid) + solve(mid,R) + h(ai,aj) (L≤i<mid,mid≤j<R)
所有现在问题的关键就在于 h(ai,aj) (L≤i<mid,mid≤j<R) 怎么求解了
对于每一个ai，我们需要在区间[mid,r)中寻找能够进位的数，所以我们可以对区间[mid,r)排序，利用二分查找找寻符合条件的最小数，然后计算值即可
时间复杂度应该是n(logn)^2
ac代码:

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<string>
#include<queue>
#include<vector>
#define ll long long

using namespace std;
const int INF = 0x3f3f3f3f;

int n,a[100005],b[9] = {10,100,1000,10000,100000,1000000,10000000,100000000,1000000000};
ll ans = 0; 
ll solve(int l,int r)
{
	ll sum = 0;
	int mid = (l+r)/2;
	if(l+1 == r) return 0;
	sum = solve(l,mid)+solve(mid,r);
	sort(a+mid,a+r);
	for(int i = l; i < mid; i++)
	{
		for(int j = 0; j < 9; j++)
		if(b[j]-a[i]>0)
		{
			int res = lower_bound(a+mid,a+r,b[j]-a[i])-a;
			sum = sum+r-res;
		}
	}
	return sum;
}
int main()
{
	scanf("%d",&n);
	for(int i = 1; i <= n; i++) scanf("%d",&a[i]);
	ll ans = solve(1,n+1);
	printf("%lld\n",ans);
	return 0;
}