【模板】splay维护序列

题目大意：维护一个长度为 N 的序列，支持单点插入，单点询问。

注意事项如下：

• build 函数中要记得初始化 fa。
• 插入两个端点值。

代码如下

#include <bits/stdc++.h>
#define pb push_back
#define mp make_pair
#define all(x) x.begin(),x.end()
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int maxn=1e5+10;

inline int read(){
    int x=0,f=1;char ch;
    do{ch=getchar();if(ch=='-')f=-1;}while(!isdigit(ch));
    do{x=x*10+ch-'0';ch=getchar();}while(isdigit(ch));
    return f*x;
}

int n,m,a[maxn];
struct node{
    #define ls(x) t[x].ch[0]
    #define rs(x) t[x].ch[1]
    int ch[2],fa,val,size;
}t[maxn<<1];
int tot,root;
inline void pushup(int o){t[o].size=t[ls(o)].size+t[rs(o)].size+1;}
inline bool get(int o){return o==rs(t[o].fa);}
inline void rotate(int o){
    int fa=t[o].fa,gfa=t[fa].fa,d1=get(o),d2=get(fa);
    t[fa].ch[d1]=t[o].ch[d1^1],t[t[o].ch[d1^1]].fa=fa;
    t[fa].fa=o,t[o].ch[d1^1]=fa;
    t[o].fa=gfa,t[gfa].ch[d2]=o;
    pushup(fa),pushup(o);
}
inline void splay(int o,int goal){
    while(t[o].fa!=goal){
        int fa=t[o].fa,gfa=t[fa].fa;
        if(gfa!=goal)get(fa)==get(o)?rotate(fa):rotate(o);
        rotate(o);
    }
    if(!goal)root=o;
}
int find(int o,int k){
    if(k<=t[ls(o)].size)return find(ls(o),k);
    else if(k>t[ls(o)].size+1)return find(rs(o),k-t[ls(o)].size-1);
    else return o;
}
int build(int fa,int l,int r){
    if(l>r)return 0;
    int o=++tot;
    int mid=l+r>>1;
    t[o].val=a[mid],t[o].fa=fa;
    ls(o)=build(o,l,mid-1),rs(o)=build(o,mid+1,r);
    return pushup(o),o;
}
void insert(int pos,int val){
    int x=find(root,pos-1),y=find(root,pos);
    splay(x,0),splay(y,x);
    ++tot,t[tot].val=val,t[tot].size=1,t[tot].fa=y,ls(y)=tot;
    pushup(y),pushup(x);
}

void read_and_parse(){
    n=m=read();
    for(int i=2;i<=n+1;i++)a[i]=read();
    root=build(0,1,n+2);
}
void solve(){
    while(m--){
        int opt=read(),l=read(),r=read(),c=read();
        if(opt==0)insert(l+1,r);
        else if(opt==1)printf("%d\n",t[find(root,r+1)].val);
    }
}
int main(){
    read_and_parse();
    solve();
    return 0;
}

转载于:https://www.cnblogs.com/wzj-xhjbk/p/10615127.html