bzoj 3786: 星系探索 dfs序+splay

题意

给出一棵树，每个节点有一个权值，要求资瓷三个操作：
C x y把x的父亲换成y
F x y把x的所有子节点的权值加上y
Q x询问x到跟的路径和
n,m<=100000

分析

用splay维护这棵树的入栈出栈序即可。
x的子树就是x的入栈点到x的出栈点
x到根的路径即使x的入栈点到根的出栈点

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<stack>
#define N 100005
#define LL long long
using namespace std;

int n,tot,sz,cnt,last[N],fa[N],w[N],ins[N*2],outs[N*2],root,a[N*2];
stack <int> sta;
struct edge{int to,next;}e[N];
struct tree{int l,r,fa;LL val,sum,w,tag;}t[N*2];
LL val[N*2];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void addedge(int u,int v)
{
    e[++cnt].to=v;e[cnt].next=last[u];last[u]=cnt;
}

void dfs(int x)
{
    a[++tot]=x;val[tot]=1;ins[x]=tot;
    for (int i=last[x];i;i=e[i].next) dfs(e[i].to);
    a[++tot]=x;val[tot]=-1;outs[x]=tot;
}

void updata(int d)
{
    t[d].sum=t[t[d].l].sum+t[t[d].r].sum+t[d].w*val[d];
    t[d].val=t[t[d].l].val+t[t[d].r].val+val[d];
}

void pushdown(int d)
{
    if (!t[d].tag) return;
    LL w=t[d].tag;t[d].tag=0;
    t[t[d].l].w+=w;t[t[d].l].tag+=w;t[t[d].l].sum+=t[t[d].l].val*w;
    t[t[d].r].w+=w;t[t[d].r].tag+=w;t[t[d].r].sum+=t[t[d].r].val*w;
}

void remove(int x,int y)
{
    for (;x!=y;x=t[x].fa) sta.push(x);
    pushdown(y);
    while (!sta.empty())
    {
        x=sta.top();sta.pop();
        pushdown(x);
    }
}

void rttr(int x)
{
    int y=t[x].l;
    t[x].l=t[y].r;
    t[t[y].r].fa=x;
    if (x==root) root=y;
    if (x==t[t[x].fa].l) t[t[x].fa].l=y;
    else if (x==t[t[x].fa].r) t[t[x].fa].r=y;
    t[y].fa=t[x].fa;
    t[x].fa=y;
    t[y].r=x;
    updata(x);updata(y);
}

void rttl(int x)
{
    int y=t[x].r;
    t[x].r=t[y].l;
    t[t[y].l].fa=x;
    if (x==root) root=y;
    if (x==t[t[x].fa].l) t[t[x].fa].l=y;
    else if (x==t[t[x].fa].r) t[t[x].fa].r=y;
    t[y].fa=t[x].fa;
    t[x].fa=y;
    t[y].l=x;
    updata(x);updata(y);
}

void splay(int x,int y)
{
    remove(x,y);
    while (t[x].fa!=y)
    {
        int p=t[x].fa,g=t[p].fa;
        if (g==y)
        {
            if (x==t[p].l) rttr(p);
            else rttl(p);
            return;
        }
        if (x==t[p].l)
        {
            if (p==t[g].l)
            {
                rttr(g);rttr(p);
            }
            else
            {
                rttr(p);rttl(g);
            }
        }
        else
        {
            if (p==t[g].r)
            {
                rttl(g);rttl(p);
            }
            else
            {
                rttl(p);rttr(g);
            }
        }
    }
}

int build(int l,int r)
{
    int mid=(l+r)/2;t[mid].w=w[a[mid]];
    if (l<mid) t[mid].l=build(l,mid-1),t[t[mid].l].fa=mid;
    if (r>mid) t[mid].r=build(mid+1,r),t[t[mid].r].fa=mid;
    updata(mid);
    return mid;
}

int findlast(int x)
{
    splay(x,0);
    x=t[x].l;int y;
    for (;x;y=x,x=t[x].r);
    return y;
}

int findnext(int x)
{
    splay(x,0);
    x=t[x].r;int y;
    for (;x;y=x,x=t[x].l);
    return y;
}

int main()
{
    n=read();
    for (int i=2;i<=n;i++)
    {
        fa[i]=read();
        addedge(fa[i],i);
    }
    for (int i=1;i<=n;i++) w[i]=read();
    tot=1;
    dfs(1);
    root=build(1,n*2+2);
    int m=read();
    while (m--)
    {
        char ch[2];
        scanf("%s",ch);
        if (ch[0]=='Q')
        {
            int x=read(),px=findlast(ins[1]),qx=findnext(ins[x]);
            splay(px,0);splay(qx,px);
            printf("%lld\n",t[t[qx].l].sum);
        }
        else if (ch[0]=='C')
        {
            int x=read(),y=read(),px=findlast(ins[x]),qx=findnext(outs[x]),py=findnext(ins[y]);
            if (fa[x]==y) continue;
            else fa[x]=y;
            splay(px,0);splay(qx,root);
            int u=t[qx].l;t[u].fa=t[qx].l=0;
            updata(qx);updata(px);
            splay(ins[y],0);splay(py,root);
            t[py].l=u;t[u].fa=py;
            updata(py);updata(ins[y]);
        }
        else
        {
            int x=read(),y=read(),px=findlast(ins[x]),qx=findnext(outs[x]);
            splay(px,0);splay(qx,root);int u=t[qx].l;
            t[u].sum+=(LL)t[u].val*y;t[u].tag+=y;t[u].w+=y;
        }
    }
    return 0;
}