萌新刚学 OI,Spaly 样例卡死求助

P3369 【模板】普通平衡树

```cpp #include<cstdio> #include<cmath> #include<algorithm> #include<queue> #include<vector> #include<cstring> #include<ctime> #include<cstdlib> #include<cctype> #include<stack> #include<map> #include<set> using namespace std; inline int read() { int s=0,w=1; char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();} while(ch>='0'&&ch<='9')s=s*10+(ch-'0'),ch=getchar(); return s*w; } inline void output(int x) { if(x/10) output(x/10); putchar(x%10+'0'); return; } const int INF=1e9; typedef long long ll; inline int Min(int x,int y){return x<y?x:y;} inline int Max(int x,int y){return x>y?x:y;} inline void Swap(int&x,int&y){x^=y;y^=x;x^=y;} const int MAXN(1e5+10); int n,m; int ch[MAXN][2],val[MAXN],cnt[MAXN],siz[MAXN]; int pool,root; inline void New(int&u,int v) { u=++pool; cnt[u]=siz[u]=1; val[u]=v; ch[u][0]=ch[u][1]=0; return; } inline void push_up(int u) { siz[u]=siz[ch[u][0]]+siz[ch[u][1]]+cnt[u]; return; } inline void zag(int&x) { int y=ch[x][1]; ch[x][1]=ch[y][0]; ch[y][0]=x; siz[y]=siz[x]; push_up(x); x=y; return; } inline void zig(int&x) { int y=ch[x][0]; ch[x][0]=ch[y][1]; ch[y][1]=x; siz[y]=siz[x]; push_up(x); x=y; return; } inline void splay(int x,int&y) { if(x==y) return; int ls=ch[y][0],rs=ch[y][1]; if(x==ls) zig(y); else if(x==rs) zag(y); else { if(val[x]<val[y]) { if(val[x]<val[ls]) splay(x,ch[ls][0]),zig(y),zig(y); else splay(x,ch[ls][1]),zag(ls),zig(y); } else { if(val[x]>val[rs]) splay(x,ch[rs][1]),zag(y),zag(y); else splay(x,ch[rs][0]),zig(rs),zag(y); } } return; } inline void ins(int&u,int v) { if(!u) New(u,v),splay(u,root); else if(val[u]==v) ++cnt[u],++siz[u],splay(u,root); else if(val[u]<v) ins(ch[u][1],v); else ins(ch[u][0],v); return; } inline void fnode(int u,int v,int&p) { if(val[u]==v){p=u;return;} if(val[u]<v) fnode(ch[u][1],v,p); else fnode(ch[u][0],v,p); } inline int findkth(int&u,int k) { if(!siz[u]) return 0; if(siz[ch[u][0]]>=k) return findkth(ch[u][0],k); else if(siz[ch[u][0]]+cnt[u]>=k) { splay(u,root); return val[root]; } return findkth(ch[u][1],k-siz[ch[u][0]]-cnt[u]); } inline void del(int&v) { int u; fnode(root,v,u); splay(u,root); if(cnt[u]>1) --siz[u],--cnt[u]; else if(!siz[ch[u][1]]) root=ch[u][0]; else { int p=findkth(ch[u][1],1); splay(p,ch[u][1]); ch[ch[u][1]][0]=ch[u][0]; root=ch[u][1]; push_up(root); } } inline int findrank(int&u,int v) { if(!siz[u]) return 0; if(val[u]==v) { int now=siz[ch[u][0]]+1; splay(u,root); return now; } if(val[u]>v) return findrank(ch[u][0],v); else return findrank(ch[u][1],v)+siz[ch[u][0]]+cnt[u]; } inline int pre(int v) { int p; fnode(root,v,p); int u=ch[p][0]; while(ch[u][1]) u=ch[u][1]; splay(u,root); return val[root]; } inline int suf(int v) { int p; fnode(root,v,p); int u=ch[p][1]; while(ch[u][0]) u=ch[u][0]; splay(u,root); return val[root]; } inline void dfs(int u,int k) { if(ch[u][0]) dfs(ch[u][0],k); printf("%d:%d\n",k,val[u]); if(ch[u][1]) dfs(ch[u][1],k); } int main() { m=read(); for(register int i=1;i<=m;i++) { int opt=read(),x=read(); if(opt==1) ins(root,x); else if(opt==2) del(x); else if(opt==3) printf("%d\n",findrank(root,x)); else if(opt==4) printf("%d\n",findkth(root,x)); else if(opt==5) printf("%d\n",pre(x)); else if(opt==6) printf("%d\n",suf(x)); // dfs(root,i); } return 0; } ```
by UperFicial @ 2021-07-08 11:12:02


《绿 勾 萌 新》
by Orange_qwq @ 2021-07-08 11:12:33


@[chensi26](/user/365825) 《红 名 萌 新》
by wang_er_gou @ 2021-07-08 11:26:07


@[princee](/user/501814) 《红 名 绿 勾 萌 新》
by Orange_qwq @ 2021-07-08 11:26:45


本蒟蒻py党无能为力
by wang_er_gou @ 2021-07-09 07:45:36


|