HDU 6534 Chika and Friendly Pairs (莫队+树状数组+离散化+预处理)

#include<bits/stdc++.h>
using namespace std;
#define rep(x,y,z) for(int (x)=(y);(x)<(z);x++)
#define mst(x,y) memset(x,y,sizeof(x))
#define ll long long
#define pii pair<int,int> 
#define piii pair<int,pii>
#define fi first
#define se second
#define mk(x,y) make_pair(x,y)
#define debug cout<<"YES"<<endl;
#define pf(x) printf("%d\n",x)
#define sc(x) scanf("%d",&x)
int gcd(int x,int y){return y?gcd(y,x%y):x;}
const int maxn=3e4+5;
int n,m,k,sz;
int tmp[maxn<<2],a[maxn],cnt=0;
struct Q{
    int l,r,k;
}q[maxn];
int res=0,ans[maxn];
///用map预处理会超时，用数组预处理不会，看题解才发现这个
int lb[maxn],ub[maxn],pos[maxn];///预处理

const int maxm=maxn<<2;细节
int bit[maxm];
void add(int x,int v){
    for(;x<maxm;bit[x]+=v,x+=x&-x);
}
int sum(int x){
    int ret=0;
    for(;x>0;ret+=bit[x],x-=x&-x);
    return ret;
}

void Add(int x){
    int tx=ub[x],ty=lb[x];
    x=pos[x];
    res+=sum(tx)-sum(ty-1);
    add(x,1);
}
void Sub(int x){
    int tx=ub[x],ty=lb[x];
    x=pos[x];
    add(x,-1);
    res-=sum(tx)-sum(ty-1);
}

int main(){
    sc(n),sc(m),sc(k);
    sz=sqrt(n);
    rep(i,1,n+1){
        sc(a[i]);
        tmp[cnt++]=a[i];
        tmp[cnt++]=a[i]-k;
        tmp[cnt++]=a[i]+k;
    }
    sort(tmp,tmp+cnt);
    cnt=unique(tmp,tmp+cnt)-tmp;
    rep(i,1,n+1){
        pos[i]=lower_bound(tmp,tmp+cnt,a[i])-tmp+1;
        lb[i]=lower_bound(tmp,tmp+cnt,a[i]-k)-tmp+1;
        ub[i]=lower_bound(tmp,tmp+cnt,a[i]+k)-tmp+1;
    }
    rep(i,0,m){
        sc(q[i].l),sc(q[i].r);
        q[i].k=i;
    }
    sort(q,q+m,[](Q x,Q y){
        return (x.l/sz==y.l/sz)?x.r<y.r:x.l/sz<y.l/sz; 
    });
    int l=1,r=0;
    rep(i,0,m){
        while(l<q[i].l) Sub(l++);
        while(l>q[i].l) Add(--l);
        while(r<q[i].r) Add(++r);
        while(r>q[i].r) Sub(r--);
        ans[q[i].k]=res;
    }
    rep(i,0,m) pf(ans[i]);
    return 0;
}