Hdu 6534 莫队+树状数组

#include <bits/stdc++.h>
using namespace std;
const int maxn = 500005;
struct node{int l,r,id;int ans;}q[maxn];
int s[maxn],Be[maxn];
int z[maxn],ni[maxn];
int unit,n,m,l=1,r;
int ans,k;
int lowbit(int x){
	return x&(-x);
}
int sum(int x){
	int as = 0;
	while(x>0){
		as+=z[x];
		x-=lowbit(x);
	}
	return as;
}
int add(int x,int d){
	while(x<=maxn){
		z[x]+=d;
		x+=lowbit(x);
	}
}
int cmp(node a,node b){
	return Be[a.l]==Be[b.l]?a.r<b.r:Be[a.l]<Be[b.l];
}
int cmp2(node a,node b){
	return a.id<b.id;
}
void revise(int x,int d){
	if(d<0)ans-=sum(x+k)-sum(x-k-1)-1,add(x,d);
	if(d>0)add(x,d),ans+=sum(x+k)-sum(x-k-1)-1;
}
int main()
{
	scanf("%d %d %d",&n,&m,&k); unit = sqrt(n);
	for(int i=1;i<=n;i++) scanf("%d",&s[i]),Be[i] = i/unit+1;
	for(int i=1;i<=m;i++){
		scanf("%d %d",&q[i].l,&q[i].r); q[i].id = i;
	}
	sort(q+1,q+1+m,cmp);
	for(int i=1;i<=m;i++){
		
		while(l<q[i].l)l++,revise(s[l-1],-1);
		while(l>q[i].l)l--,revise(s[l],1);
		while(r<q[i].r)r++,revise(s[r],1);
		while(r>q[i].r)r--,revise(s[r+1],-1);

	      q[i].ans = ans;
	}
	sort(q+1,q+1+m,cmp2);
	for(int i=1;i<=m;i++){
	 	printf("%d\n",q[i].ans);
	}
	return 0;
 } 
 /*
 6
 1 4 7 5 6 3
 4
 1 4
 2 5
 4 5
 4 6
 */