poj 2418
#include<iostream> #include<string.h> #include<stdio.h> #include<string> #include <stddef.h> using namespace std; int cnt=0; #define abs(x) (((x)<0)?(-x):(x)) typedef string Int; struct AVLNode{ AVLNode *chi[2],*pa; Int val; int bf,cnt; }*nil=new AVLNode(),*root=nil; inline void Nil(){ nil->chi[0]=nil->chi[1]=nil->pa=nil; nil->bf=nil->cnt=0;} inline AVLNode * CreatNode(AVLNode * &node,Int val,int cnt=1){ if(node!=NULL && node!=nil){ AVLNode *p=node; delete p; } node=new AVLNode(); node->val=val; node->cnt=cnt; node->pa=node->chi[0]=node->chi[1]=nil; node->bf=0; return node; } inline void Connect(AVLNode *pa,AVLNode *node,int dir) { pa->chi[dir]=node; node->pa=pa; } AVLNode * Bound(AVLNode * node,Int val){ if(node==nil) return nil; for(int dir;val!=node->val;node=node->chi[dir]){ dir=(val<node->val)?0:1; if(node->chi[dir]==nil) return node; } return node; } AVLNode * Extrema(AVLNode * node,int dir){ while(node->chi[dir]!=nil) node=node->chi[dir]; return node; } AVLNode * Cessor(AVLNode * node,int dir){ if(node->chi[dir]!=nil) return Extrema(node->chi[dir],1-dir); for(AVLNode * p=node->pa;p!=nil && p->chi[dir]==node;node=p,p=p->pa); return node->pa; } void Rotate(AVLNode *node,int rodir){ Nil(); AVLNode *pivot=node->chi[1-rodir]; Connect(node,node->chi[1-rodir]->chi[rodir],1-rodir); Connect(node->pa,pivot,(node->pa->chi[0]==node)?0:1); Connect(pivot,node,rodir); Nil(); node->bf+=(rodir)?2:-2; pivot->bf+=(rodir)?1:-1; } AVLNode * Search(AVLNode *root,Int val){ AVLNode *p=Bound(root,val); if(p->val==val) return p; return nil; } void Balance(AVLNode * node,int dbf,int aimbf){ for((node->bf)+=dbf;aimbf!=abs(node->bf);(node->bf)+=dbf){ if(abs(node->bf)==2){ int dir=(node->bf<0)?0:1; if((node->bf)*(node->chi[dir]->bf)<0) Rotate(node->chi[dir],dir); Rotate(node,1-dir); node->pa->bf-=dbf; } if((node=node->pa)==nil) return; } } void Insert(AVLNode * &root,Int val){ if(root==nil){ root=CreatNode(root,val); return; } Nil(); AVLNode *p=Bound(root,val); if(p->val==val){ p->cnt++; return; } int dir=(val<p->val)?0:1; AVLNode *node=nil; CreatNode(node,val); Connect(p,node,dir); Balance(p,(!dir)?-1:1,0); } bool Delete(AVLNode *root,Int val){ if(root==nil) return false; Nil(); AVLNode *p=Search(root,val); if(p==nil) return false; if(p->cnt>1){ (p->cnt)--; return true; } if(p->chi[0]!=nil || p->chi[1]!=nil){ AVLNode *suc=Cessor(p,1); if(suc==nil) suc=Cessor(p,0); p->val=suc->val; p=suc; } int dir=p->pa->chi[0]==p?0:1; p->pa->chi[dir]=nil; Balance(p->pa,(!dir)?1:-1,1); delete p; return true; } void Traversal(AVLNode * root){ if(root->chi[0]!=nil) Traversal(root->chi[0]); cout<<root->val; printf(" %.4lf\n",(double)root->cnt/(double)cnt); // double if(root->chi[1]!=nil) Traversal(root->chi[1]); } int main(){ char * s; while (scanf("%s",s)!=EOF && ++cnt) Insert(root,s); // getline? Traversal(root); return 0; }
// A.myc #include<iostream> #include<string.h> #include<stdio.h> #include<string> #include <stddef.h> #include<queue> using namespace std; int oo=0xffffffff; int cnt=0; #define abs(x) (((x)<0)?(-x):(x)) typedef string Int; struct AVLNode{ AVLNode *chi[2],*pa; Int val; int bf,cnt; int blank,len,hei; }*nil=new AVLNode(),*root=nil; void OutNode(AVLNode * node){ for(int i=0;i<node->blank;i++) printf(" "); cout<<node->val; } void CalCor(AVLNode *root,int cor,int &lst,int &rst,int dir){ root->len=root->val.length(); cor-=dir*root->len/2; if(root->chi[0]!=nil){ CalCor(root->chi[0],cor,lst,root->blank,-1); root->chi[0]->hei=root->hei+1;} if(lst<0){ cor-=lst; rst-=lst; root->blank-=lst; lst=0; } cor+=root->len; if(cor>rst){ rst=cor; } if(root->chi[1]!=nil){ CalCor(root->chi[1],cor,lst,root->blank,1); root->chi[1]->hei=root->hei+1; } } void OutTree(AVLNode * root){ queue<AVLNode *> q; q.push(root); int y=0; while(!q.empty()){ AVLNode* t=q.front(); if(t->hei!=y) { y=t->hei; printf("\n"); } OutNode(t); if(t->chi[0]!=nil) q.push(t->chi[0]); if(t->chi[1]!=nil) q.push(t->chi[1]); q.pop(); } } void Doit(){ int l=0; root->hei=0; CalCor(root,0,l,oo,-1); OutTree(root); printf("\n//////////////////////////////////\n"); } inline void Nil(){ nil->chi[0]=nil->chi[1]=nil->pa=nil; nil->bf=nil->cnt=0;} inline AVLNode * CreatNode(AVLNode * &node,Int val,int cnt=1){ if(node!=NULL && node!=nil){ AVLNode *p=node; delete p; } node=new AVLNode(); node->val=val; node->cnt=cnt; node->pa=node->chi[0]=node->chi[1]=nil; node->bf=0; return node; } inline void Connect(AVLNode *pa,AVLNode *node,int dir) { pa->chi[dir]=node; node->pa=pa; } AVLNode * Bound(AVLNode * node,Int val){ if(node==nil) return nil; for(int dir;val!=node->val;node=node->chi[dir]){ dir=(val<node->val)?0:1; if(node->chi[dir]==nil) return node; } return node; } AVLNode * Extrema(AVLNode * node,int dir){ while(node->chi[dir]!=nil) node=node->chi[dir]; return node; } AVLNode * Cessor(AVLNode * node,int dir){ if(node->chi[dir]!=nil) return Extrema(node->chi[dir],1-dir); for(AVLNode * p=node->pa;p!=nil && p->chi[dir]==node;node=p,p=p->pa); return node->pa; } void Rotate(AVLNode *node,int rodir){ Nil(); AVLNode *pivot=node->chi[1-rodir]; Connect(node,node->chi[1-rodir]->chi[rodir],1-rodir); Connect(node->pa,pivot,(node->pa->chi[0]==node)?0:1); Connect(pivot,node,rodir); Nil(); node->bf+=(rodir)?2:-2; pivot->bf+=(rodir)?1:-1; Doit(); } AVLNode * Search(AVLNode *root,Int val){ AVLNode *p=Bound(root,val); if(p->val==val) return p; return nil; } void Balance(AVLNode * node,int dbf,int aimbf){ for((node->bf)+=dbf;aimbf!=abs(node->bf);(node->bf)+=dbf){ if(abs(node->bf)==2){ int dir=(node->bf<0)?0:1; if((node->bf)*(node->chi[dir]->bf)<0) Rotate(node->chi[dir],dir); Rotate(node,1-dir); node->pa->bf-=dbf; } if((node=node->pa)==nil) return; } } void Insert(AVLNode * &root,Int val){ if(root==nil){ root=CreatNode(root,val); return; } Nil(); AVLNode *p=Bound(root,val); if(p->val==val){ p->cnt++; return; } int dir=(val<p->val)?0:1; AVLNode *node=nil; CreatNode(node,val); Connect(p,node,dir); Balance(p,(!dir)?-1:1,0); Doit(); } bool Delete(AVLNode *root,Int val){ if(root==nil) return false; Nil(); AVLNode *p=Search(root,val); if(p==nil) return false; if(p->cnt>1){ (p->cnt)--; return true; } if(p->chi[0]!=nil || p->chi[1]!=nil){ AVLNode *suc=Cessor(p,1); if(suc==nil) suc=Cessor(p,0); p->val=suc->val; p=suc; } int dir=p->pa->chi[0]==p?0:1; p->pa->chi[dir]=nil; Balance(p->pa,(!dir)?1:-1,1); delete p; Doit(); return true; } void Traversal(AVLNode * root){ if(root->chi[0]!=nil) Traversal(root->chi[0]); cout<<root->val; printf(" %.4lf\n",(double)root->cnt/(double)cnt*100.0); // double if(root->chi[1]!=nil) Traversal(root->chi[1]); } int main(){ freopen("out.txt","w",stdout); string s; while (getline(cin,s)) // ## if(s!="" && ++cnt) Insert(root,s); // ## Traversal(root); return 0; } /*#include<iostream> #include<string.h> #include<stdio.h> #include<string> #include <stddef.h> #include<queue> using namespace std; int oo=0xffffffff; int cnt=0; #define abs(x) (((x)<0)?(-x):(x)) typedef string Int; struct AVLNode{ AVLNode *chi[2],*pa; Int val; int bf,cnt; int blank,len; }*nil=new AVLNode(),*root=nil; void CalCor(AVLNode *root,int cor,int &lst,int &rst,int dir){ root->len=root->val.length(); cor-=dir*root->len/2; if(root->chi[0]!=nil) CalCor(root->chi[0],cor,lst,root->blank,-1); if(lst<0){ cor-=lst; rst-=lst; root->blank-=lst; lst=0; } cor+=root->len; if(cor>rst){ rst=cor; } if(root->chi[1]!=nil) CalCor(root->chi[1],cor,lst,root->blank,1); } void OutTree(AVLNode * root){ queue<AVLNode *> q; q.push(root); while(!q.empty()){ AVLNode* t=q.front(); q.push(t->chi[0]); q.push(t->chi[1]); q.pop(); } } void Doit(){ int l=0; CalCor(root,0,l,oo,-1); OutTree(root); } inline void Nil(){ nil->chi[0]=nil->chi[1]=nil->pa=nil; nil->bf=nil->cnt=0;} inline AVLNode * CreatNode(AVLNode * &node,Int val,int cnt=1){ if(node!=NULL && node!=nil){ AVLNode *p=node; delete p; } node=new AVLNode(); node->val=val; node->cnt=cnt; node->pa=node->chi[0]=node->chi[1]=nil; node->bf=0; return node; } inline void Connect(AVLNode *pa,AVLNode *node,int dir) { pa->chi[dir]=node; node->pa=pa; } AVLNode * Bound(AVLNode * node,Int val){ if(node==nil) return nil; for(int dir;val!=node->val;node=node->chi[dir]){ dir=(val<node->val)?0:1; if(node->chi[dir]==nil) return node; } return node; } AVLNode * Extrema(AVLNode * node,int dir){ while(node->chi[dir]!=nil) node=node->chi[dir]; return node; } AVLNode * Cessor(AVLNode * node,int dir){ if(node->chi[dir]!=nil) return Extrema(node->chi[dir],1-dir); for(AVLNode * p=node->pa;p!=nil && p->chi[dir]==node;node=p,p=p->pa); return node->pa; } void Rotate(AVLNode *node,int rodir){ Nil(); AVLNode *pivot=node->chi[1-rodir]; Connect(node,node->chi[1-rodir]->chi[rodir],1-rodir); Connect(node->pa,pivot,(node->pa->chi[0]==node)?0:1); Connect(pivot,node,rodir); Nil(); node->bf+=(rodir)?2:-2; pivot->bf+=(rodir)?1:-1; Doit(); } AVLNode * Search(AVLNode *root,Int val){ AVLNode *p=Bound(root,val); if(p->val==val) return p; return nil; } void Balance(AVLNode * node,int dbf,int aimbf){ for((node->bf)+=dbf;aimbf!=abs(node->bf);(node->bf)+=dbf){ if(abs(node->bf)==2){ int dir=(node->bf<0)?0:1; if((node->bf)*(node->chi[dir]->bf)<0) Rotate(node->chi[dir],dir); Rotate(node,1-dir); node->pa->bf-=dbf; } if((node=node->pa)==nil) return; } } void Insert(AVLNode * &root,Int val){ if(root==nil){ root=CreatNode(root,val); return; } Nil(); AVLNode *p=Bound(root,val); if(p->val==val){ p->cnt++; return; } int dir=(val<p->val)?0:1; AVLNode *node=nil; CreatNode(node,val); Connect(p,node,dir); Balance(p,(!dir)?-1:1,0); Doit(); } bool Delete(AVLNode *root,Int val){ if(root==nil) return false; Nil(); AVLNode *p=Search(root,val); if(p==nil) return false; if(p->cnt>1){ (p->cnt)--; return true; } if(p->chi[0]!=nil || p->chi[1]!=nil){ AVLNode *suc=Cessor(p,1); if(suc==nil) suc=Cessor(p,0); p->val=suc->val; p=suc; } int dir=p->pa->chi[0]==p?0:1; p->pa->chi[dir]=nil; Balance(p->pa,(!dir)?1:-1,1); delete p; return true; Doit(); } void Traversal(AVLNode * root){ if(root->chi[0]!=nil) Traversal(root->chi[0]); cout<<root->val; printf(" %.4lf\n",(double)root->cnt/(double)cnt); // double if(root->chi[1]!=nil) Traversal(root->chi[1]); } int main(){ char * s; while (scanf("%s",s)!=EOF && ++cnt) Insert(root,s); // getline? Traversal(root); return 0; }*/
// AVLTree #include<iostream> #include<string.h> #include<stdio.h> #include<string> #include<stddef.h> #include<queue> #include<fstream> using namespace std; #define abs(x) (((x)<0)?(-x):(x)) #define BLACK 0 #define RED 1 #define debug typedef string Int; typedef int Val; int cnt=0; struct RBNode{ RBNode *chi[2],*pa; int color,padir; Int key; Val cnt; #ifdef debug int blank,hei,len; #endif RBNode * s(){ return pa->chi[1-padir]; } }*nil=new RBNode(),*root=nil; void Nil(){ nil->chi[0]=nil->chi[1]=nil->pa=nil; nil->color=nil->cnt=nil->padir=0; nil->key="";} #ifdef debug typedef RBNode AVLNode; // PaintAVL void OutNode(AVLNode * node,int cor){ for(int i=cor;i<node->blank;i++) printf(" "); cout<<node->key; } int CalCor(AVLNode * root,int lst){ Nil(); root->blank=0; root->len=root->key.length(); int dis=0,dis2=0;; if(root->chi[0]!=nil){ root->chi[0]->hei=root->hei+1; dis=CalCor(root->chi[0],lst); } int halflen=(root->len)/2; root->blank=lst+dis-halflen; if(root->blank<lst) root->blank=lst; if(root->chi[1]!=nil){ root->chi[1]->hei=root->hei+1; dis2=CalCor(root->chi[1],lst+dis+(dis?1:0)); } if(!dis2) root->blank-=(root->len+1)/2; if(root->blank<lst) root->blank=lst; int ans=((dis+(dis?1:0)+dis2)>root->len)?dis+(dis?1:0)+dis2:root->len; if(dis2 && dis && (root->chi[1]->blank-root->chi[0]->blank)>root->len && (root->chi[0]->blank+root->chi[1]->blank)/2+root->len<=ans) root->blank=(root->chi[0]->blank+root->chi[1]->blank)/2; return ans; Nil(); } void OutTree(AVLNode * root){ queue<AVLNode *> q; q.push(root); int y=0,cor=0,cor2=0,cor3=0; string branch="",pro=""; while(!q.empty()){ AVLNode* t=q.front(); if(t->hei!=y) { cout<<endl<<pro<<endl<<branch; branch=""; pro=""; y=t->hei; printf("\n"); cor=0; cor2=0; cor3=0; } OutNode(t,cor); cor=t->blank+t->len; for(int i=cor3;i<t->blank;i++) pro+=' '; cor3=t->blank; switch(t->color){ case 1: pro+="RED "; cor3+=5; break; case 0: pro+="BLACK "; cor3+=7; break; } if(t->chi[0]!=nil){ q.push(t->chi[0]); for(int i=cor2;i<(t->chi[0]->blank+t->blank)/2;i++) branch+=' '; branch+='/'; cor2=(t->chi[0]->blank+t->blank)/2; } if(t->chi[1]!=nil){ q.push(t->chi[1]); for(int i=cor2;i<(t->chi[1]->blank+t->blank)/2;i++) branch+=' '; branch+='\\'; cor2=(t->chi[1]->blank+t->blank)/2; } q.pop(); } cout<<endl<<pro; } void Doit(Int key){ cout<<key<<endl<<endl; root->hei=0; CalCor(root,0); OutTree(root); printf("\n//////////////////////////////////\n"); } #endif RBNode * CreatNode(RBNode * &node,Int key,Val cnt=1,int op=1){ if(!op){ return nil; } if(node!=NULL && node!=nil){ RBNode *p=node; delete p; } node=new RBNode(); node->cnt=cnt; node->key=key; node->color=RED; node->pa=node->chi[0]=node->chi[1]=nil; return node; } void Connect(RBNode *pa,RBNode *node,int dir) { Nil(); pa->chi[dir]=node; node->pa=pa; node->padir=dir; Nil(); #ifdef debug Doit("Connect "+pa->key+" "+node->key+(dir?" 1":" 0")); #endif } RBNode * Extrema(RBNode * node,int dir){ while(node->chi[dir]!=nil) node=node->chi[dir]; return node; } RBNode * Cessor(RBNode * node,int dir){ if(node->chi[dir]!=nil) return Extrema(node->chi[dir],1-dir); for(RBNode * p=node->pa;p!=nil && p->chi[dir]==node;node=p,p=p->pa); return node->pa; } RBNode * Bound(RBNode * node,Int key,int dif=1){ if(node==nil) return nil; for(int dir;key!=node->key;node=node->chi[dir]){ dir=(key<node->key)?0:1; if(node->chi[dir]==nil) return node; } if(!dif && node->key!=key) return nil; return node; } void SwpClr(RBNode * a,RBNode * b) { int t=a->color; a->color=b->color; b->color=t; #ifdef debug Doit("Swap "+a->key+" "+b->key); #endif } void Rotate(RBNode *node,int rodir,int wis=1){ #ifdef debug Doit("Before Ro "+node->key+(rodir?" 1 ":" 0 ")); #endif Nil(); if(wis && node->chi[1-rodir]->chi[rodir]!=nil && node->chi[1-rodir]->chi[1-rodir]->color==BLACK) Rotate(node->chi[1-rodir],1-rodir); while(node->chi[rodir]->chi[1-rodir]!=nil) Rotate(node->chi[rodir],rodir); RBNode * pivot=node->chi[1-rodir]; SwpClr(node,pivot); Connect(node,node->chi[1-rodir]->chi[rodir],1-rodir); Connect(node->pa,pivot,node->padir); Connect(pivot,node,rodir); if(root==node) root=pivot; Nil(); #ifdef debug Doit("After Ro "+node->key+(rodir?" 1 ":" 0 ")); #endif } bool InDe(RBNode * &root,Int key,int op,Val cnt=1){ #ifdef debug Doit("In "+key); #endif Nil(); RBNode * pa=Bound(root,key,op); RBNode * node=new RBNode(); node=CreatNode(node,key,cnt,op); node->padir=(node==nil || node->key<pa->key)?0:1; if(root==nil){ if(!op) return false; root=node; root->color=BLACK; Nil(); return true;} if(pa->chi[0]!=nil && pa->chi[1]!=nil){ RBNode * t=Cessor(pa,1); pa->key=t->key; pa->cnt=t->cnt; pa=t; } if(op && pa->color!= node->color){ Connect(pa,node,node->padir); return true; }; if(!op && pa->chi[0]!=nil) Rotate(pa,0); Connect(pa,node,node->padir); node=pa; while(node->s()->color==RED || node->s()!=nil && (node->color==node->s()->chi[0]->color==node->s()->chi[1]->color==node->pa->color==BLACK)){ if(node->s()->color!=node->color){ Rotate(pa,pa->padir); continue; } (node->s()->color)^=1; SwpClr(node,node->pa); node=node->pa; if(node->pa->color!=node->color){ if(op) node=node->pa; break; } } if(node->color==RED || node->s()->chi[0]->color==RED || node->s()->chi[1]->color==RED) { Rotate(node->pa,node->padir^(op-BLACK)); } else SwpClr(node,node->pa); root->color=BLACK; return true; } void Traversal(RBNode * root){ if(root->chi[0]!=nil) Traversal(root->chi[0]); cout<<root->cnt; printf(" %.4lf\n",(double)root->cnt/(double)cnt*100.0); if(root->chi[1]!=nil) root->color=BLACK; Traversal(root->chi[1]); } int main(){ freopen("in.txt","r",stdin); freopen("out.txt","w",stdout); string s; while (getline(cin,s)) if(s!="" && ++cnt){ InDe(root,s,1); // if(root!=nil){ InDe(root,s,0); InDe(root,s,1);} } Traversal(root); return 0; }