首页 > 代码库 > 【BZOJ-3786】星系探索 Splay + DFS序
【BZOJ-3786】星系探索 Splay + DFS序
3786: 星系探索
Time Limit: 40 Sec Memory Limit: 256 MBSubmit: 647 Solved: 212
[Submit][Status][Discuss]
Description
物理学家小C的研究正遇到某个瓶颈。
他正在研究的是一个星系，这个星系中有n个星球，其中有一个主星球(方便起见我们默认其为1号星球)，其余的所有星球均有且仅有一个依赖星球。主星球没有依赖星球。
我们定义依赖关系如下：若星球a的依赖星球是b,则有星球a依赖星球b.此外，依赖关系具有传递性，即若星球a依赖星球b,星球b依赖星球c,则有星球a依赖星球c.
对于这个神秘的星系中，小C初步探究了它的性质，发现星球之间的依赖关系是无环的。并且从星球a出发只能直接到达它的依赖星球b.
每个星球i都有一个能量系数wi.小C想进行若干次实验，第i次实验，他将从飞船上向星球di发射一个初始能量为0的能量收集器，能量收集器会从星球di开始前往主星球，并收集沿途每个星球的部分能量，收集能量的多少等于这个星球的能量系数。
但是星系的构成并不是一成不变的，某些时刻，星系可能由于某些复杂的原因发生变化。
有些时刻，某个星球能量激发，将使得所有依赖于它的星球以及他自己的能量系数均增加一个定值。还有可能在某些时刻，某个星球的依赖星球会发生变化，但变化后依然满足依赖关系是无环的。
现在小C已经测定了时刻0时每个星球的能量系数,以及每个星球(除了主星球之外)的依赖星球。接下来的m个时刻，每个时刻都会发生一些事件。其中小C可能会进行若干次实验，对于他的每一次实验，请你告诉他这一次实验能量收集器的最终能量是多少。
Input
第一行一个整数n,表示星系的星球数。
接下来n-1行每行一个整数,分别表示星球2-n的依赖星球编号。
接下来一行n个整数，表示每个星球在时刻0时的初始能量系数wi.
接下来一行一个整数m,表示事件的总数。
事件分为以下三种类型。
(1)"Q di"表示小C要开始一次实验,收集器的初始位置在星球di.
(2)"C xi yi"表示星球xi的依赖星球变为了星球yi.
(3)"F pi qi"表示星球pi能量激发，常数为qi.
Output
对于每一个事件类型为Q的事件，输出一行一个整数，表示此次实验的收集器最终能量。
Sample Input
1
1
4 5 7
5
Q 2
F 1 3
Q 2
C 2 3
Q 2
Sample Output
15
25
HINT
n<=100000,m<=300000,1<di,xi<=n,wi,qi<=100000.保证操作合法。
Source
By 佚名上传
Solution
子树有关修改，显然不能用Link-Cut-Tree...
考虑用DFS序，但换根操作显然不能用线段树维护，所以用支持分裂合并的Splay！
我们把DFS序中的入栈出栈分开，在一个点入栈时权值为+val[x]，出栈时权值为-val[x]，这样对于他的子树，影响显然会抵消
这样的话，查询点x到跟的$\sum_{x}^{root}val[x]$，就相当于前缀和查询$\sum_{pl[1]}^{pl[x]}val[x]$
子树修改，就是打标记。
换根操作，把x换到y，先把x的子树区间分裂开，再把y的子树区间提取，然后把x的子树区间连到y的子树区间最左即可
Code
#include<iostream>#include<cstdio>#include<cmath>#include<algorithm>#include<cstring>#include<stack>using namespace std;inline int read(){ int x=0; char ch=getchar(); while (ch<‘0‘ || ch>‘9‘) {ch=getchar();} while (ch>=‘0‘ && ch<=‘9‘) {x=x*10+ch-‘0‘; ch=getchar();} return x;}#define LL long long#define MAXN 200010 int N,M,power[MAXN],val[MAXN];struct EdgeNode{int next,to,d;}edge[MAXN<<1];int head[MAXN],cnt=1;inline void AddEdge(int u,int v) {cnt++; edge[cnt].next=head[u]; head[u]=cnt; edge[cnt].to=v;}inline void InsertEdge(int u,int v) {AddEdge(u,v); AddEdge(v,u);}int pl[MAXN],dfn,pr[MAXN],pre[MAXN],del[MAXN];void DFS(int now,int last){ pl[now]=++dfn; pre[dfn]=now; del[dfn]=1; for (int i=head[now]; i; i=edge[i].next) if (edge[i].to!=last) DFS(edge[i].to,now); pr[now]=++dfn; del[dfn]=-1;}namespace SplayTree{ int fa[MAXN],son[MAXN][2],sz,root,ln[MAXN],rn[MAXN],pos[MAXN]; LL sum[MAXN],tag[MAXN]; //ln记录左的数量，rn记录右的数量，tag标记 #define ls(x) son[x][0] #define rs(x) son[x][1] #define INF 1000100 inline void Update(int now) { sum[now]=sum[ls(now)]+sum[rs(now)]+(LL)val[now]; ln[now]=ln[ls(now)]+ln[rs(now)]+(del[now]==1); rn[now]=rn[ls(now)]+rn[rs(now)]+(del[now]==-1); } inline void paint(int now,LL D) {if (now) val[now]+=D*del[now],sum[now]+=D*(ln[now]-rn[now]),tag[now]+=D;} inline void PushDown(int now) { if (!tag[now]) return; LL D=tag[now]; tag[now]=0; if (ls(now)) paint(ls(now),D); if (rs(now)) paint(rs(now),D); } void print(int now) { PushDown(now); if (val[now]!=INF && val[now]!=-INF) printf("ID=%d ---%d [%d %d] %d %I64d %d %d %d %I64d\n", now,fa[now],ls(now),rs(now),val[now],sum[now],ln[now],rn[now],del[now],tag[now]); if (son[now][0]) print(son[now][0]); if (son[now][1]) print(son[now][1]); } inline bool Right(int now) {return son[fa[now]][1]==now;} inline void rotate(int now) { PushDown(fa[now]); PushDown(now); int f=fa[now],gf=fa[f],wh=Right(now); son[f][wh]=son[now][wh^1]; fa[son[f][wh]]=f; fa[f]=now; son[now][wh^1]=f; fa[now]=gf; if (gf) son[gf][son[gf][1]==f]=now; Update(f); Update(now); } inline void splay(int now,int tar) { for (int f; (f=fa[now])!=tar; rotate(now)) if (fa[f]!=tar) rotate(Right(now)==Right(f)? f:now); if (!tar) root=now; } inline int fmin(int x) {while (son[x][0]) x=son[x][0]; return x;} inline int fmax(int x) {while (son[x][1]) x=son[x][1]; return x;} inline void Split(int l,int r) { splay(l,0); int x=fmax(ls(root)); splay(r,0); int y=fmin(rs(root)); splay(x,0); splay(y,x); } inline LL Query(int pos) {Split(pl[1],pl[pos]); return sum[ls(rs(root))];} inline void Change(int x,int D) { Split(pl[x],pr[x]); int rt=son[rs(root)][0]; sum[rt]+=(LL)(ln[rt]-rn[rt])*D; val[rt]+=D*del[rt]; tag[rt]+=D; Update(rs(root)); Update(root); } inline void BuildTree(int l,int r,int last) { if (r<l) return; int mid=(l+r)>>1,now=mid; fa[now]=last; son[last][now>last]=now; if (l==r) {sum[now]=val[now]; ln[now]=del[now]==1; rn[now]=1-ln[now]; return;} BuildTree(l,mid-1,now); BuildTree(mid+1,r,now); Update(now); } inline void Init() {BuildTree(1,(N<<1)+2,0); root=(1+(N<<1)+2)>>1;} inline void MakeRoot(int now,int father) { Split(pl[now],pr[now]); int rt=ls(rs(root)); fa[ls(rs(root))]=0; son[rs(root)][0]=0; Update(rs(root)),Update(root); splay(pl[father],0); splay(fmin(rs(root)),root); son[rs(root)][0]=rt; fa[rt]=rs(root); Update(rs(root)),Update(root); }}using namespace SplayTree;int main(){ N=read(); for (int i=2; i<=N; i++) InsertEdge(i,read()); dfn=1; DFS(1,0); for (int i=1; i<=N; i++) power[i]=read();// for (int i=1; i<=N; i++) printf("id=%d [%d , %d]\n",pre[i],pl[pre[i]],pr[pre[i]]); for (int i=1; i<=N; i++) val[pl[i]]+=power[i],val[pr[i]]-=power[i];// for (int i=1; i<=dfn; i++) printf("%d ",val[i]); puts(""); M=read(); SplayTree::Init(); while (M--) {// print(root); char opt[5]; scanf("%s",opt); int x,y; if (opt[0]==‘Q‘) x=read(),printf("%I64d\n",SplayTree::Query(x)); if (opt[0]==‘C‘) x=read(),y=read(),SplayTree::MakeRoot(x,y); if (opt[0]==‘F‘) x=read(),y=read(),SplayTree::Change(x,y); } return 0;}
一天前..我：char哥，要是DFS序不用线段树，用Splay的话，不就可以支持子树交换XXX了吗... char哥：...woc！这么科学....
于是跟char哥连坐，写了一天Splay，然后这道题昨天没调出来...
终于写出来了...40s时限41s AC....
我：我一定要压到40s以内！ ..... 我：当我没说；
我：char哥，你有什么压常数的技巧么？ char哥：重写，你这程序已经没救了！......
本机加O2优化：
【BZOJ-3786】星系探索 Splay + DFS序