刷题记录:牛客NC14380位数差

传送门:牛客

题目描述:

输入:
10
0 1 2 3 4 5 6 7 8 9
输出:
20

主要思路:
这是一道典型的分治题,然而当时我是在做二分的题集里碰到这道题的,然后脑子里就一直在想该怎么二分答案,想来想去感觉复杂度都没有降低,没想到是一道分治题(笑哭)

1. 首先对于函数h来说,将其分成两半对于最终答案显然是没有影响的,因为其只需找到两个数字即可,所以我们可以有以下的关系solve(l,r)=solve(l,mid)+solve(mid+1,r)+h(l,r),因为solve求出的两边自己的位数差,缺少了恰好分在两边的位数差值,此时我们将其求出即可
2. 得出了可以使用分治的思想之后,我们就可以考虑该如何计算位数差了,这个感觉还是需要一定的思维的,刚开始我只能想到取模?之类的方法,感觉复杂度都挺高的,应该会被T掉,之后发现了一个比较巧妙的转化,举一个栗子

我们有B=1234,此时我们想让A+B能够进位,显然我们需要A>=10000-1234(进位+1)或者A>=100000-1234(进位+2),\
是不是已经找到规律了,
没错我们只要不断的去让10的次方去每次减去B的值,再判断有几个A大于此值即可,注意此时我们不需要判断加几个,
因为越大的数字是会被反复叠加的
查找的方法我们可以使用二分,但是为了方便,我这里采用了c++STL中的lower_bound函数,
lower_bound()返回值是一个迭代器,返回指向大于等于key的第一个值的位置,当然它的函数内部实现形式也是二分,因此十分优秀

下面是具体代码:

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
#define ll_maxn 0x3f3f3f3f3f3f3f3f
ll n;ll a[maxn];ll ans=0;
ll fuzhu[maxn]={10,100,1000,10000,100000,1000000,10000000,100000000,1000000000};
ll solve(ll l,ll r) {
	if(l==r) return 0;
	ll mid=(l+r)>>1;
	ans=solve(l,mid)+solve(mid+1,r);
	sort(a+mid+1,a+r+1);
	for(int i=l;i<=mid;i++) {
		for(int j=0;j<=8;j++) {
			if(fuzhu[j]-a[i]>0) {
				ll pos=lower_bound(a+mid+1,a+r+1,fuzhu[j]-a[i])-a;
				ans+=(r-pos+1);
			}
		}
	}
	return ans;
}
int main(){
	n=read();
	for(int i=1;i<=n;i++) a[i]=read();
	cout<<solve(1,n)<<endl;;
	
	return 0;
}