首页 > 代码库 > BZOJ 3786 星系探索 Splay维护树的入栈出栈序
BZOJ 3786 星系探索 Splay维护树的入栈出栈序
题目大意:给出一棵树,要求有以下这些操作:1.求出一个节点到根的点权和。2.将一个节点的父亲改变。3.将一个子树中的每一个节点都加上一个权值。
思路:LCT就不用想了,因为有子树操作。然后就是一个很神奇的东西了,就是Splay维护树的入栈出栈序。这个玩应是做了这个题之后才知道的。但是感觉真的很dio。
首先,我们先按照题意,将树建出来。然后从根开始深搜,这样一个点进入DFS函数和出DFS函数的时候就会有两个时间点,就是入栈的时间和出栈的时间。然后利用Splay维护一个序列,就是入栈出栈的顺序。在数组中入栈存正,出栈存负。这个序列有很优美的性质,每一个节点的子树就是这个点的入栈和出栈之间加的东西。两点之间的路径就是一个点的入栈到另一个点的入栈。
然后这个题就是代码题了。。
CODE:
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #define MAX 600010 #define WORKPATH (root->son[1]->son[0]) using namespace std; struct SplayTree{ long long val,sum,c,size; long long plus,num,_num; SplayTree *son[2],*father; SplayTree(int _,int __); SplayTree() {} bool Check() { return father->son[1] == this; } void Combine(SplayTree *a,bool dir) { son[dir] = a; a->father = this; } void Plus(int c); void PushUp() { sum = son[0]->sum + son[1]->sum + val * (plus ? 1:-1); size = son[0]->size + son[1]->size + 1; num = son[0]->num + son[1]->num + (plus == 1); _num = son[0]->_num + son[1]->_num + (plus == 0); } void PushDown() { if(c) { son[0]->Plus(c); son[1]->Plus(c); c = 0; } } }none,*nil = &none,*root,*tree[MAX]; SplayTree:: SplayTree(int _,int __) { plus = __; if(__ == 1) ++num; if(__ == 0) ++_num; val = _; sum = _ * (plus ? 1:-1); size = 1; c = 0; son[0] = son[1] = nil; } void SplayTree:: Plus(int _) { if(this == nil) return ; sum += _ * (num - _num); val += _; c += _; } int points,asks; int head[MAX],total; int next[MAX],aim[MAX]; int src[MAX],pos[MAX]; int from[MAX],cnt; int p[MAX]; char c[10]; inline void Add(int x,int y) { next[++total] = head[x]; aim[total] = y; head[x] = total; } void DFS(int x) { pos[++cnt] = src[x]; from[cnt] = x; p[cnt] = true; for(int i = head[x]; i; i = next[i]) DFS(aim[i]); pos[++cnt] = src[x]; from[cnt] = x; p[cnt] = false; } void Pretreatment() { nil->son[0] = nil->son[1] = nil->father = nil; nil->val = nil->sum = nil->c = nil->size = 0; p[0] = p[cnt + 1] = -1; } SplayTree *BuildTree(int l,int r) { if(l > r) return nil; int mid = (l + r) >> 1; SplayTree *re = new SplayTree(pos[mid],p[mid]); if(p[mid]) tree[from[mid] << 1] = re; else tree[from[mid] << 1|1] = re; re->Combine(BuildTree(l,mid - 1),false); re->Combine(BuildTree(mid + 1,r),true); re->PushUp(); return re; } inline void Rotate(SplayTree *a,bool dir) { SplayTree *f = a->father; f->PushDown(),a->PushDown(); f->son[!dir] = a->son[dir]; f->son[!dir]->father = f; a->son[dir] = f; f->father->son[f->Check()] = a; a->father = f->father; f->father = a; f->PushUp(); if(root == f) root = a; } inline void Splay(SplayTree *a,SplayTree *aim) { while(a->father != aim) { if(a->father->father == aim) Rotate(a,!a->Check()); else if(!a->father->Check()) { if(!a->Check()) Rotate(a->father,true),Rotate(a,true); else Rotate(a,false),Rotate(a,true); } else { if(a->Check()) Rotate(a->father,false),Rotate(a,false); else Rotate(a,true),Rotate(a,false); } } a->PushUp(); } SplayTree *Find(SplayTree *a,int k) { a->PushDown(); if(a->son[0]->size >= k) return Find(a->son[0],k); k -= a->son[0]->size; if(k == 1) return a; return Find(a->son[1],k - 1); } inline void SplaySeg(SplayTree *a,SplayTree *b) { int size_1,size_2; Splay(a,nil); size_1 = root->son[0]->size + 1; Splay(b,nil); size_2 = root->son[0]->size + 1; Splay(Find(root,size_1 - 1),nil); Splay(Find(root,size_2 + 1),root); } int main() { cin >> points; for(int x,i = 2; i <= points; ++i) { scanf("%d",&x); Add(x,i); } for(int i = 1; i <= points; ++i) scanf("%d",&src[i]); DFS(1); Pretreatment(); root = BuildTree(0,cnt + 1); root->father = nil; cin >> asks; for(int x,y,i = 1; i <= asks; ++i) { scanf("%s",c); if(c[0] == 'Q') { scanf("%d",&x); SplaySeg(tree[1 << 1],tree[x << 1]); printf("%lld\n",WORKPATH->sum); } else if(c[0] == 'C') { scanf("%d%d",&x,&y); SplaySeg(tree[x << 1],tree[x << 1|1]); SplayTree *temp = WORKPATH; WORKPATH = nil; root->son[1]->PushUp(); root->PushUp(); Splay(tree[y << 1],nil); int k = root->son[0]->size + 1; Splay(Find(root,k + 1),root); root->son[1]->Combine(temp,false); root->son[1]->PushUp(); root->PushUp(); } else { scanf("%d%d",&x,&y); SplaySeg(tree[x << 1],tree[x << 1|1]); WORKPATH->Plus(y); } } return 0; }
BZOJ 3786 星系探索 Splay维护树的入栈出栈序
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。