【bzoj3489】 A simple rmq problem k-d树

由于某些原因，我先打了一个错误的树套树，后来打起了$k-d$。接着因不明原因在思路上被卡了很久，在今天中午蹲坑时恍然大悟......

对于一个数字$a_i$，我们可以用一组三维坐标$(i,pre,nxt)$来表示，其中$i$表示该数字下标，$pre$表示在区间$[1,i)$中满足$a[j]=a[i]$的最大$j$，若不存在，则$pre=0$。$nxt$表示在区间$(i,n]$中满足$a[j]=a[i]$的最小$j$，若不存在，则$nxt=n+1$。

接着我们种一棵3-d树去存储这n个点。对于任意一个节点，需存储该节点的数字，以该节点为根的字数中最大的数字，每一维的最小值和最大值。对于一组询问$[l,r]$，答案即为满足第一维在区间$[l,r]$，第二维在区间$[0,l)$，第三维在区间$(r,n+1)$中的所有点上的最大数字。在查询时，直接按照$k-d$的正常查询策略更新答案即可。

时间复杂度:$O(n log n+n^\frac{5}{3})$

警告：若采用此方法，此题建议各位加一个微小的剪枝：假定当前所得到的最大答案为$ans$，若当前访问的字树中最大答案$≤ans$，直接跳过这棵字树即可。（借助该方法极限数据耗时从17.5s降低至1.4s）

#include<bits/stdc++.h>
#define M 210000
using namespace std;
int n,m,D,root;
struct kd{
    int a[3],max[3],min[3],now,ans,l,r;
    kd(){a[0]=a[1]=a[2]=now=l=r=0;min[1]=min[2]=min[0]=12345678;}
    kd(int xx,int yy,int zz,int kk){a[0]=xx;a[1]=yy;a[2]=zz;now=kk;}
    friend bool operator <(kd a,kd b){
        return a.a[D]<b.a[D];
    }
}a[M];
void upd(kd &x,kd &y){
    for(int i=0;i<3;i++)
        x.max[i]=max(x.max[i],y.max[i]),
        x.min[i]=min(x.min[i],y.min[i]);
}
inline int nx(int d){if(d==2) return 0; return d+1;}
void build(int &x,int l,int r,int d){
    if(l>r) return; int mid=(l+r)>>1;
    D=d; nth_element(a+l,a+mid,a+r+1); x=mid; 
    for(int i=0;i<3;i++) a[x].max[i]=a[x].min[i]=a[x].a[i];    
    build(a[x].l,l,mid-1,nx(d));
    build(a[x].r,mid+1,r,nx(d));
    upd(a[x],a[a[x].l]); upd(a[x],a[a[x].r]);
    a[x].ans=max(a[x].now,max(a[a[x].l].ans,a[a[x].r].ans));
}
int ans=0;
void query(int x,int l,int r){
    if(x==0||ans>=a[x].ans||r<a[x].min[0]||a[x].max[0]<l||l-1<a[x].min[1]||a[x].max[2]<r+1) return;
    if(l<=a[x].min[0]&&a[x].max[0]<=r&&a[x].max[1]<l&&r<a[x].min[2])
    {ans=max(ans,a[x].ans); return;}
    if(l<=a[x].a[0]&&a[x].a[0]<=r&&a[x].a[1]<l&&r<a[x].a[2]) ans=max(ans,a[x].now); 
    query(a[x].l,l,r); query(a[x].r,l,r);
}
int last[M]={0};
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        int x; scanf("%d",&x);
        a[i].a[0]=i; a[i].now=x;
        a[i].a[1]=last[x];
        a[last[x]].a[2]=i;
        last[x]=i;
    }
    for(int i=1;i<=n;i++) if(a[i].a[2]==0) a[i].a[2]=n+1;
    build(root,1,n,0);
    while(m--){
        int x,y,l,r; scanf("%d%d",&x,&y);
        l=(x+ans)%n+1; r=(y+ans)%n+1; if(l>r) swap(l,r);
        ans=0; query(root,l,r);
        printf("%d\n",ans);
    }
}