HDU 4638 线段树离线操作

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int a[maxn];
struct node{
    int l,r,v;
}tree[maxn<<2];
struct pp{
    int l,r,id;
}po[maxn];
int pre[maxn],ans[maxn];
bool cmp(pp x,pp y){
    return x.r<y.r;
}
void build(int l,int r,int i){
    tree[i].l=l,tree[i].r=r,tree[i].v=0;
    if(l==r)return;
    int mid=l+r>>1;
    build(l,mid,i<<1);
    build(mid+1,r,i<<1|1);
}
void update(int pos,int rt,int d){
    tree[rt].v+=d;
    if(tree[rt].l==tree[rt].r){return;}
    int mid=tree[rt].l+tree[rt].r>>1;
    if(pos<=mid)
        update(pos,rt<<1,d);
    else
        update(pos,rt<<1|1,d);
}
int query(int l,int r,int rt){
    if(tree[rt].l==l&&tree[rt].r==r){
        return tree[rt].v;
    }
    int mid=tree[rt].l+tree[rt].r>>1;
    if(r<=mid){
        query(l,r,rt<<1);
    }
    else if(l>=mid+1){
        query(l,r,rt<<1|1);
    }
    else{
        return (query(l,mid,rt<<1)+query(mid+1,r,rt<<1|1));
    }
}
int main(){
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    int t;cin>>t;
    while(t--){
        int n,m;cin>>n>>m;
        for(int i=1;i<=n;i++){
            cin>>a[i];
        }
        for(int i=0;i<m;i++){
            cin>>po[i].l>>po[i].r;
            po[i].id=i;
        }
        sort(po,po+m,cmp);
//        for(int i=0;i<m;i++)cout<<po[i].l<<" "<<po[i].r<<" "<<po[i].id<<endl;
        build(1,n,1);
        memset(pre,-1,sizeof(pre));
        for(int i=1,j=0;i<=n&&j<m;i++){
            pre[a[i]]=i;
            update(i,1,1);
            if(pre[a[i]+1]!=-1)update(pre[a[i]+1],1,-1);
            if(pre[a[i]-1]!=-1)update(pre[a[i]-1],1,-1);

            while(j<m&&i==po[j].r){
//                cout<<po[j].id<<"asd"<<endl;
                ans[po[j].id]=query(po[j].l,po[j].r,1);
                j++;
            }
        }
        for(int i=0;i<m;i++){
            cout<<ans[i]<<endl;
        }
    }
}