BZOJ 3786 星系探索 Splay维护DFS序

题目大意:给定一棵有根树,有三种操作:询问结点到根的权值和,修改结点的父亲,子树加值。
由于有子树修改,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;
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值