(2019徐州网络赛) I. query (树状数组)

传送门

题意：一个1~n的全排列，有m次查询，查询[l,r]中满足（l<=i<j<=r）a[i]和a[j]是倍数关系的点对。

解：我们用s[i]表示带第i个，前面总共用s[i]个满足条件的点对，考虑[l,r]就是s[r]-s[l-1]吗，显然还多考虑了那种一个点在[1,l-1],一个点在[l,r]里的这种点对，我们需要将这种减掉。从第一个点往后考虑，那么可以和这个点构成点对的，显然就是它的倍数或者它的因子，当然不包括它本身，我们只加入pos位大于现在i位的，避免重复添加。当qu[j].l==i时，我们先不加入a[i],而是考虑减去已经出现在[l,r]中的点对，这不就是我们对于这qu[j]查询要减掉的吗，然后再加入a[i]的贡献；而对于q[j].r==i,当然是先考虑a[i]的贡献，然后加上[l,r]中的点对。

#include<bits/stdc++.h>
#define il inline
#define pb push_back
#define ms(_data,v) memset(_data,v,sizeof(_data))
#define SZ(a) int((a).size())
#define low(x) (x&(-x))
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f;
const int N=1e5+5;
//il int Add(ll &x,ll y) {return x=x+y>=mod?x+y-mod:x+y;}
//il int Mul(ll &x,ll y) {return x=x*y>=mod?x*y%mod:x*y;}
ll s[N],ans[N];
il void add(int p,int x){
	while(p<N) s[p]+=x,p+=low(p);
}
il ll ask(int p){
	ll res=0;
	while(p) res+=s[p],p-=low(p);
	return res;
}
il ll r_ask(int l,int r){
	return ask(r)-ask(l-1);
} 
int pos[N],n,m,a[N];
struct node{
	int l,r,id;
}qu[N];
il bool cmpl(node a,node b){
	if(a.l==b.l) return a.r<b.r;
	else return a.l<b.l; 
}
il bool cmpr(node a,node b){
	if(a.r==b.r) return a.l<b.l;
	else return a.r<b.r;
}
vector<int> v[N];
void init(){
	for(int i=2;i<N;++i){
		for(ll j=2;j*j<=i;++j){
			if(i%j==0){
				if(j*j==i) v[i].pb(j);
				else v[i].pb(j),v[i].pb(i/j);
			}
		}
		v[i].pb(1);
	}
}
int main(){
//	std::ios::sync_with_stdio(0);cin.tie(0);
	init();
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;++i){
		scanf("%d",a+i);
		pos[a[i]]=i;
	}
	for(int i=1;i<=m;++i){
		scanf("%d%d",&qu[i].l,&qu[i].r);
		qu[i].id=i;
	}
	sort(qu+1,qu+m+1,cmpl);
	int j=1;
	for(int i=1;i<=n;++i){
		while(qu[j].l==i && j<=m){
			ans[qu[j].id]-=r_ask(qu[j].l,qu[j].r);
			j++;
		}
		for(ll k=2;k*a[i]<=n;++k){
			if(pos[k*a[i]]>i) add(pos[k*a[i]],1);
		}
		for(auto t:v[a[i]]){
			if(pos[t]>i) add(pos[t],1);
		}
	}
	ms(s,0);
	sort(qu+1,qu+m+1,cmpr);
	j=1;
	for(int i=1;i<=n;++i){
		for(ll k=2;k*a[i]<=n;++k){
			if(pos[k*a[i]]>i) add(pos[k*a[i]],1);
		}
		for(auto t:v[a[i]]){
			if(pos[t]>i) add(pos[t],1);
		}
		while(qu[j].r==i && j<=m){
			ans[qu[j].id]+=r_ask(qu[j].l,qu[j].r);
			j++;
		}
	}
	for(int i=1;i<=m;++i){
		printf("%lld\n",ans[i]);
	}
	return 0;
}