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 }
zoj3765Lights(splay),布布扣,bubuko.com
时间： 2024-09-29 20:46:45