首页 > 代码库 > [TYVJ1728/BZOJ3224]普通平衡树-替罪羊树

[TYVJ1728/BZOJ3224]普通平衡树-替罪羊树

Problem 普通平衡树

Solution

本题是裸的二叉平衡树。有很多种方法可以实现。这里打的是替罪羊树模板。 
此题极其恶心。 
前驱后继模块需要利用到rank模块来换一种思路求。 
很多细节的地方容易炸。我拿数据调了很久才A。 
(delt()删除模块其实是不需要重建的,不影响时间复杂度) 
替罪羊树具体详见此篇知乎:替罪羊树

AC Code

  1 #include "iostream"
  2 #include "cstdio"
  3 #define alpha 0.7
  4 using namespace std;
  5 struct ScapeGoatTree{
  6     int lc,rc,n,w,s,fa;
  7 }a[100010];
  8 struct REbuild{
  9     int n,w;
 10 }d[100010];
 11 void rebuild(int p);
 12 int tot=1,tott=0,top=0,h[100010],n,T,X,root;
 13 int check(int p){
 14     if(a[p].w==0)return 0;
 15     if ((a[a[p].lc].s+a[a[p].lc].w>=double(alpha*(a[p].w+a[p].s)))||
 16         (a[a[p].rc].s+a[a[p].rc].w>=double(alpha*(a[p].w+a[p].s))))return 1;
 17     return 0;
 18 }
 19 int getN(int x){
 20     int now=root;
 21     while(a[now].n!=x&&a[now].s!=0){
 22         if(a[now].n>=x)now=a[now].lc;
 23         else now=a[now].rc;
 24     }
 25     return now;
 26 }
 27 void insr(int x,bool rb){
 28     if(root==0){
 29         root=1;
 30         a[1].n=x;
 31         a[1].w++;
 32         return;
 33     }
 34     int now=root;
 35     while(a[now].s!=0){
 36         if(x==a[now].n){
 37             a[now].w++;
 38             return;
 39         }
 40         if(x>a[now].n&&a[now].rc==0)break;
 41         if(x<a[now].n&&a[now].lc==0)break;
 42         a[now].s++;
 43         if(x>a[now].n)now=a[now].rc;
 44         else if(x<a[now].n)now=a[now].lc;
 45     }
 46 
 47     if(x==a[now].n){
 48         a[now].w++;
 49         return;
 50     }
 51     a[now].s++;
 52     int tmp=now;
 53     if(x>a[now].n)now=a[now].rc=++tot;
 54     else if(x<a[now].n)now=a[now].lc=++tot;
 55     a[now].w++;
 56     a[now].n=x;
 57     if(tmp!=now)a[now].fa=tmp;
 58     int chk=0;
 59     while(now!=root){
 60         now=a[now].fa;
 61         if(check(now))chk=now;
 62     }
 63     if(rb)rebuild(chk);
 64 }
 65 int getnewp(){
 66     if(top>0){
 67         top--;
 68         return h[top+1];
 69     }
 70     else return ++tot;
 71 }
 72 void dfs(int p){
 73     h[++top]=p;
 74     if(a[p].lc!=0)dfs(a[p].lc);
 75     if(a[p].w!=0)d[++tott].n=a[p].n,
 76     d[tott].w=a[p].w;
 77     a[p].s=0;
 78     if(a[p].rc!=0)dfs(a[p].rc);
 79 }
 80 void make(int l,int r,int p,int fa){
 81     int mid=(l+r)/2;
 82     a[p].fa=fa;
 83     a[p].w=d[mid].w;
 84     a[p].n=d[mid].n;
 85     if(mid-1>=l)a[p].lc=getnewp(),make(l,mid-1,a[p].lc,p);else a[p].lc=0;
 86     if(mid+1<=r)a[p].rc=getnewp(),make(mid+1,r,a[p].rc,p);else a[p].rc=0;
 87     a[p].s=a[a[p].lc].w+a[a[p].lc].s+a[a[p].rc].w+a[a[p].rc].s;
 88 }
 89 void rebuild(int p){
 90     if(p==0)return;
 91     tott=0;
 92     dfs(p);
 93     int now=getnewp();
 94     a[now].fa=a[p].fa;
 95     if(p==root)root=now;
 96     int mid=(1+tott)/2;
 97     if(d[mid].n>a[a[p].fa].n)a[a[p].fa].rc=now;
 98     else a[a[p].fa].lc=now;
 99     make(1,tott,now,a[p].fa);
100 }
101 void delt(int p,bool rb){
102     a[p].w--;
103     int chk=0;
104     while(p!=root){
105         p=a[p].fa;
106         a[p].s--;
107         if(check(p))chk=p;
108     }
109 //  if(rb)rebuild(chk);
110 }
111 int XgetRk(int x);
112 int RkgetX(int x);
113 int suc(int x){
114     insr(x+1,0);
115     int tmp=XgetRk(x+1),nx=getN(x+1);
116     delt(nx,0);
117     return RkgetX(tmp);
118 }
119 int pre(int x){
120     insr(x,0);
121     int tmp=XgetRk(x);
122     delt(getN(x),0);
123     return RkgetX(tmp-1);
124 }
125 int XgetRk(int x){
126     int p=getN(x),ans;
127     ans=a[a[p].lc].s+a[a[p].lc].w;
128     while(p!=root){
129         if(a[a[p].fa].rc==p)ans+=a[a[a[p].fa].lc].s+a[a[a[p].fa].lc].w+a[a[p].fa].w;
130         p=a[p].fa;
131     }
132     return ans+1;
133 }
134 int RkgetX(int x){
135     int now=root;
136     while(true){
137         int lcS=a[a[now].lc].s+a[a[now].lc].w;
138         if(x<=lcS)now=a[now].lc;
139         else if(x>lcS&&x<=a[now].w+lcS)return a[now].n;
140         else x-=lcS+a[now].w,now=a[now].rc;
141         if(x==0)return a[now].fa;
142     }
143 }
144 int main(){
145     scanf("%d",&n);
146     for(int i=1;i<=n;i++){
147         if(i%1000==0){
148             int ttott;
149             ttott++;
150         }
151         scanf("%d%d",&T,&X);                                                        
152         if(T==1)insr(X,1);  
153         else if(T==2)delt(getN(X),1);
154         else if(T==3)printf("%d\n",XgetRk(X));
155         else if(T==4)printf("%d\n",RkgetX(X));
156         else if(T==5)printf("%d\n",pre(X));
157         else if(T==6)printf("%d\n",suc(X)); 
158     }   
159 }

 

[TYVJ1728/BZOJ3224]普通平衡树-替罪羊树