BZOJ 3685 普通van Emde Boas树

来源：互联网发布：java在线人数统计编辑：程序博客网时间：2024/05/05 09:08

题目链接：233333

一棵裸裸的线段树

rank为一个点在现在里面的排名

第一个和第二个操作脑补成单点修改

第三个和第四个操作脑补成看rank求点

第五个和第六个操作脑补成求一个点的rank数然后+1或-1转为看rank求点

第七个操作===你不会我也没办法

程序中的rank可以理解为sum，自己手贱没办法

#include <cstdio>#include <iostream>#include <cstring>#include <algorithm>#define rep(j,k,l) for (int j=k;j<=l;j++)#define N 1000005using namespace std;struct _233{int l,r,rank;} tr[N*5];int a[N],cnt;int read(){char ch=getchar();int s=0;while(ch>'9'||ch<'0') ch=getchar();while(ch<='9'&&ch>='0'){s=s*10+ch-'0';ch=getchar();}return s;}void stree(int k,int l,int r){if (l==r) return;tr[k].l=++cnt;tr[k].r=++cnt;stree(tr[k].l,l,(l+r)/2);stree(tr[k].r,(l+r)/2+1,r);}void add(int k,int l,int r,int p){if (l==r){tr[k].rank=1;return;}int mid=(l+r)/2;if (p<=mid) add(tr[k].l,l,(l+r)/2,p);else add(tr[k].r,(l+r)/2+1,r,p);tr[k].rank=tr[tr[k].l].rank+tr[tr[k].r].rank;}void del(int k,int l,int r,int p){if (l==r){tr[k].rank=0;return;}int mid=(l+r)/2;if (p<=mid) del(tr[k].l,l,(l+r)/2,p);else del(tr[k].r,(l+r)/2+1,r,p);tr[k].rank=tr[tr[k].l].rank+tr[tr[k].r].rank;}int ask_rank(int k,int l,int r,int p){if (l==r) return l;if (p<=tr[tr[k].l].rank) return ask_rank(tr[k].l,l,(l+r)/2,p);else return ask_rank(tr[k].r,(l+r)/2+1,r,p-tr[tr[k].l].rank);}int ask(int k,int l,int r,int o,int p){if (r<o||l>p) return 0;if (o<=l&&r<=p) return tr[k].rank;return ask(tr[k].l,l,(l+r)/2,o,p)+ask(tr[k].r,(l+r)/2+1,r,o,p);}int main(){int n,T;n=read();T=read();cnt=1;stree(1,1,n);while (T--){int k=read();if (k==1){int l=read();l++;a[l]=1;add(1,1,n,l);}else if (k==2){int l=read();l++;a[l]=0;del(1,1,n,l);}else if (k==3){if (tr[1].rank==0) printf("-1\n");else printf("%d\n",ask_rank(1,1,n,1)-1);}else if (k==4){if (tr[1].rank==0) printf("-1\n");else printf("%d\n",ask_rank(1,1,n,tr[1].rank)-1);}else if (k==5){int l=read();l++;int p=ask(1,1,n,1,l);if (a[l]==0) p++;if (p==1) printf("-1\n");else printf("%d\n",ask_rank(1,1,n,p-1)-1);}else if (k==6){int l=read();l++;int p=ask(1,1,n,1,l);if (p==tr[1].rank) printf("-1\n");else printf("%d\n",ask_rank(1,1,n,p+1)-1);}else{int l=read();l++;printf("%d\n",a[l]?1:-1);}}system("pause");}

0 0