题目大意:给定一棵有根树,有三种操作:询问结点到根的权值和,修改结点的父亲,子树加值。
由于有子树修改,LCT用不了啦。还有什么解决树上问题的方法呢?看看能不能转化成序列问题吧。
记录一下入栈出栈序,结点权值入栈存正出栈存负,结点到根的权值和即DFS序中从1加到该点入栈的位置。由于是DFS序,所以子树在序列中一定是连在一起的,这样修改父亲和子树加值都不用愁啦,整体移动/打一个加标记就好。动态序列问题就找Splay。
以为挺简单的,然而足足耗了一天…我就是想偷个懒直接权值正负代表入栈出栈结果还有0啊啊啊!!!因为没找出来这毛病我写了三遍这题233
(Split、Merge函数感觉好怪异然而不写函数的话真的辣眼睛)
#include <cstdio>
#include <cstring>
#define N 100005
using namespace std;
typedef long long LL;
struct Point {
int ord,val;
bool flag;
Point(int x=0,int y=0,bool z=false):ord(x),val(y),flag(z){}
}seq[N<<1];
struct Node {
Node *ch[2],*pa;
int s,p;
LL sum,add,v;
bool flag;
int dir() { return (pa->ch[0]==this) ? 0 : (pa->ch[1]==this) ? 1 : -1;}
void maintain() {
ch[0]->pushdown(); ch[1]->pushdown();
s=ch[0]->s+ch[1]->s+1;
sum=ch[0]->sum+ch[1]->sum+v;
p=ch[0]->p+ch[1]->p+(flag?1:-1);
return ;
}
void pushdown();
Node(int x=0,bool y=false);
}*null=new Node(),*root=null,*pos[N][2];
Node :: Node(int x,bool y):v(x),flag(y) {s=null ? 1 : 0; ch[0]=ch[1]=pa=null; sum=v; p=null ? (flag?1:-1) : 0; add=0;}
void Node :: pushdown() {
if(!add || this==null) return ;
if(!flag) v-=add;
else v+=add;
sum+=p*add;
if(ch[0]!=null) ch[0]->add+=add;
if(ch[1]!=null) ch[1]->add+=add;
add=0;
return ;
}
struct Edge {
int to,nex;
}e[N];
int T,fir[N];
LL val[N];
void dfs(int x,int las) {
seq[++T]=Point(x,val[x],true);
for(int i=fir[x];i!=-1;i=e[i].nex) {
if(e[i].to==las) continue;
dfs(e[i].to,x);
}
seq[++T]=Point(x,-val[x],false);
return ;
}
void init(Node*& o,int l,int r) {
if(l>r) return ;
int mid=l+r>>1;
if(o==null) o=new Node(seq[mid].val,seq[mid].flag);
if(seq[mid].flag) pos[seq[mid].ord][0]=o;
else pos[seq[mid].ord][1]=o;
init(o->ch[0],l,mid-1); init(o->ch[1],mid+1,r);
if(o->ch[0]!=null) o->ch[0]->pa=o;
if(o->ch[1]!=null) o->ch[1]->pa=o;
o->maintain();
return ;
}
void Rotate(Node* o,int d) {
Node* k=o->ch[d^1]; int d2;
o->ch[d^1]=k->ch[d]; k->ch[d]->pa=o;
k->ch[d]=o;
o->maintain(); k->maintain();
if(~(d2=o->dir())) o->pa->ch[d2]=k;
k->pa=o->pa; o->pa=k;
return ;
}
void To_pushdown(Node* o) {
if(o->pa!=null) To_pushdown(o->pa);
o->pushdown();
return ;
}
void Splay(Node* o) {
int d; To_pushdown(o);
while(~(d=o->dir())) {
if(o->pa->dir()==d) Rotate(o->pa->pa,d^1);
Rotate(o->pa,d^1);
}
return ;
}
Node* Find_R(Node* o) {
if(o->ch[1]!=null) return Find_R(o->ch[1]);
return o;
}
void Split(Node* o,int d,Node*& l,Node*& r) {
Splay(o);
if(!d) l=o->ch[0] , r=o;
else l=o , r=o->ch[1];
o->ch[d]->pa=null; o->ch[d]=null;
o->maintain();
return ;
}
Node* Merge(Node* l,Node* r) {
if(l==null) return r;
Node* o=Find_R(l);
Splay(o);
o->ch[1]=r;
if(r!=null) r->pa=o;
o->maintain();
return o;
}
void DFS(Node* o) {
o->pushdown();
if(o->ch[0]!=null) DFS(o->ch[0]) , o->ch[0]->pa=o;
printf("%d ",o->v);
if(o->ch[1]!=null) DFS(o->ch[1]) , o->ch[1]->pa=o;
return ;
}
int main() {
null->ch[0]=null->ch[1]=null->pa=null;
int n,m;
scanf("%d",&n);
memset(fir,-1,sizeof fir);
for(int i=0;i<n-1;i++) {
int x;
scanf("%d",&x);
e[i].to=i+2; e[i].nex=fir[x]; fir[x]=i;
}
for(int i=1;i<=n;i++) scanf("%lld",&val[i]) , pos[i][0]=pos[i][1]=null;
dfs(1,0);
init(root,1,n<<1);
scanf("%d",&m);
while(m--) {
char mode[2];
int x,y;
scanf("%s",mode);
if(mode[0]=='Q') {
scanf("%d",&x);
Node *l,*r;
Split(pos[x][0],1,l,r);
printf("%lld\n",l->sum);
root=Merge(l,r);
}
if(mode[0]=='C') {
scanf("%d%d",&x,&y);
Node *l,*mid,*r,*o;
Split(pos[x][0],0,l,mid);
Split(pos[x][1],1,mid,r);
root=Merge(l,r);
Split(pos[y][0],1,l,r);
root=Merge(l,Merge(mid,r));
}
if(mode[0]=='F') {
scanf("%d%d",&x,&y);
Node *l,*mid,*r;
Split(pos[x][0],0,l,mid);
Split(pos[x][1],1,mid,r);
mid->add+=y;
root=Merge(l,Merge(mid,r));
}
root->pa=null;
}
return 0;
}