zoj3765Lights(splay)

最新推荐文章于 2018-08-17 09:48:24 发布
weixin_30879169
最新推荐文章于 2018-08-17 09:48:24 发布
阅读量82
点赞数
原文链接：http://www.cnblogs.com/shangyu/p/3796717.html
版权
链接
splay的增删改操作。
刚开始对于某段区间首先有了lazy标记时，把其左右孩子给交换了，导致在pushup时又交换了一次而debug了n久。
  1 #include <iostream>
  2 #include<cstdio>
  3 #include<cstring>
  4 #include<algorithm>
  5 #include<stdlib.h>
  6 #include<vector>
  7 #include<cmath>
  8 #include<queue>
  9 #include<set>
 10 using namespace std;
 11 #define N 300010
 12 #define LL long long
 13 #define INF 0xfffffff
 14 #define key_value ch[ch[root][1]][0]
 15 const double eps = 1e-8;
 16 const double pi = acos(-1.0);
 17 const double inf = ~0u>>2;
 18 using namespace std;
 19 int a[N][2];
 20 struct splay_tree
 21 {
 22     int pre[N],size[N],vv[N];
 23     int ch[N][2];
 24     int root,n,tot,num,tot1;
 25     int s[N][2],sta[N],key[N];
 26     void dfs(int x)
 27     {
 28         if(x)
 29         {
 30             dfs(ch[x][0]);
 31             printf("结点%2d:左儿子 %2d 右儿子 %2d 父结点 %2d size=%2d,key=%2d s0=%d  s1 = %d vv=%d\n",
 32                    x,ch[x][0],ch[x][1],pre[x],size[x],key[x],s[x][0],s[x][1],vv[x]);
 33             dfs(ch[x][1]);
 34         }
 35     }
 36     void debug()
 37     {
 38         printf("root:%d\n",root);
 39         dfs(root);
 40     }
 41 //以上用于debug*/
 42     void newnode(int &x,int v,int sta,int fa)//新建一结点
 43     {
 44         x = ++tot;
 45         ch[x][0]=ch[x][1] = 0;
 46         pre[x] = fa;
 47         s[x][0] = s[x][1] = 0;
 48         s[x][sta] = v;
 49         size[x] = 1;
 50         vv[x] = sta;
 51         key[x] = v;
 52     }
 53     void pushup(int w)//由儿子更新其父亲
 54     {
 55         size[w] = size[ch[w][0]]+size[ch[w][1]]+1;
 56         int st = vv[w];
 57         int l = ch[w][0],r = ch[w][1];
 58         int cnt1 = 0,cnt2 = 0;
 59         cnt1 = s[l][0];
 60         if(s[r][0])
 61         {
 62             if(cnt1) cnt1 = __gcd(s[r][0],cnt1);
 63             else cnt1 = s[r][0];
 64         }
 65         cnt2 = s[l][1];
 66         if(s[r][1])
 67         {
 68             if(cnt2) cnt2 = __gcd(s[r][1],cnt2);
 69             else cnt2 = s[r][1];
 70         }
 71         s[w][0] = cnt1,s[w][1] = cnt2;
 72         s[w][st] = s[w][st]?__gcd(s[w][st],key[w]):key[w];
 73         //cout<<s[w][0]<<" "<<s[w][1]<<endl;
 74     }
 75     void rotate(int r,int kind)//旋转操作，根据kind进行左旋和右旋
 76     {
 77         int y = pre[r];
 78         ch[y][!kind] = ch[r][kind];
 79         pre[ch[r][kind]] = y;
 80         if(pre[y])
 81         {
 82             ch[pre[y]][ch[pre[y]][1]==y] = r;
 83         }
 84         pre[r] = pre[y];
 85         ch[r][kind] = y;
 86         pre[y] = r;
 87         pushup(y);
 88         pushup(r);
 89     }
 90     void splay(int r,int goal)//将r结点旋至goal下
 91     {
 92         while(pre[r]!=goal)
 93         {
 94             if(pre[pre[r]]==goal)
 95             {
 96                 rotate(r,ch[pre[r]][0]==r);
 97             }
 98             else
 99             {
100                 int y = pre[r];
101                 int kind = (ch[pre[y]][0]==y);
102                 if(ch[y][kind]==r)
103                 {
104                     rotate(r,!kind);
105                     rotate(r,kind);
106                 }
107                 else
108                 {
109                     rotate(y,kind);
110                     rotate(r,kind);
111                 }
112             }
113         }
114         pushup(r);
115         if(goal==0) root = r;
116     }
117     int get_k(int k)//得到第k个的结点
118     {
119         int r = root;
120         while(size[ch[r][0]]+1!=k)
121         {
122             if(size[ch[r][0]]>=k)
123                 r = ch[r][0];
124             else
125             {
126                 k-=(size[ch[r][0]]+1);//根据左右结点的数量来确定第k个节点在哪里
127                 r = ch[r][1];
128             }
129         }
130         pushup(r);
131         return r;
132     }
133     void add(int k,int val,int sta)//添加一个结点 位置自己修改 这里为第一个
134     {
135         splay(get_k(k),0);
136         splay(get_k(k+1),root);
137         int r = ch[root][1];
138         newnode(ch[r][0],val,sta,r);
139         pushup(r);
140         pushup(root);
141     }
142     void updelete(int r)//删除第r个结点
143     {
144         splay(get_k(r),0);
145         splay(get_k(r+2),root);
146         key_value = 0;
147         pushup(ch[root][1]);
148         pushup(root);
149     }
150     void update(int k,int flag,int v) //更新区间信息
151     {
152         splay(get_k(k),0);
153         splay(get_k(k+2),root);
154         if(flag)
155         {
156             vv[key_value]^=1;
157             swap(s[key_value][1],s[key_value][0]);
158         }
159         else
160         {key[key_value] = v;s[key_value][vv[key_value]] = v;}
161         pushup(ch[root][1]);
162         pushup(root);
163 
164     }
165     int query(int l,int r,int sta)//询问l,r区间，将第l-1个结点旋自根，第r+1个结点旋自根的有儿子，
166     {
167         //则l-r就变成了根的右儿子的左儿子
168         splay(get_k(l),0);
169         splay(get_k(r+2),root);
170         return s[key_value][sta];
171     }
172     void build(int &x,int l,int r,int fa)
173     {
174         int m = (l+r)>>1;
175         if(l>r) return ;
176         newnode(x,a[m][0],a[m][1],fa);
177         build(ch[x][0],l,m-1,x);
178         build(ch[x][1],m+1,r,x);
179         pushup(x);
180     }
181     void init(int o)
182     {
183         int i;
184         for(i = 1; i <= o; i++)
185         scanf("%d%d",&a[i][0],&a[i][1]);
186         size[0] = ch[0][0] = ch[0][1] = s[0][0] = s[0][1] = key[0] = vv[0] = 0;
187         root = tot = 0;
188         newnode(root,0,0,0);
189         newnode(ch[root][1],0,0,root);
190         build(ch[ch[root][1]][0],1,o,ch[root][1]);
191         size[root] = 2;
192         pushup(ch[root][1]);
193         pushup(root);
194     }
195 }SP;
196 int main()
197 {
198     int n,q;
199     while(scanf("%d%d",&n,&q)!=EOF)
200     {
201         SP.init(n);
202         while(q--)
203         {
204             char sq[20];
205             int k,x,y;
206             scanf("%s%d",sq,&k);
207             if(sq[0]=='I')
208             {
209                 scanf("%d%d",&x,&y);
210                 SP.add(k+1,x,y);
211                // SP.debug();
212             }
213             else if(sq[0]=='D')
214             {
215                 SP.updelete(k);
216             }
217             else if(sq[0]=='R')
218             {
219                 SP.update(k,1,0);
220             }
221             else if(sq[0]=='M')
222             {
223                 scanf("%d",&x);
224                 SP.update(k,0,x);
225             }
226             else if(sq[0]=='Q')
227             {
228                 scanf("%d%d",&x,&y);
229                 int ans = SP.query(k,x,y);
230                 if(ans==0)
231                 printf("-1\n");
232                 else
233                 printf("%d\n",ans);
234                // SP.debug();
235             }
236         }
237     }
238     return 0;
239 }