[BZOJ 2759] 一个动态树好题

[BZOJ 2759] 一个动态树好题

题目描述

首先这是个基环树。

然后根节点一定会连出去一条非树边。通过一个环就可以解除根的答案，然后其他节点的答案就可以由根解出来。

因为要修改\(p_i\)，所以我们用\(lct\)。

还是有点难写的。

代码：

#include<bits/stdc++.h>
#define ll long long
#define N 30005
#define ls ch[v][0]
#define rs ch[v][1]

using namespace std;
inline int Get() {int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}while('0'<=ch&&ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;}

const ll mod=10007;
ll inv[mod];
int n;
int k[N],b[N];
int p[N];
int fa[N],ch[N][2];
int rev[N];
int A[N],B[N];
int val[N];
void update(int v) {
    A[v]=(A[ls]*k[v])%mod;
    B[v]=(B[ls]*k[v]+b[v])%mod;
    A[v]=(A[v]*A[rs])%mod;
    B[v]=(B[v]*A[rs]+B[rs])%mod;
}

bool isroot(int v) {return v!=ch[fa[v]][0]&&v!=ch[fa[v]][1];}

void rot(int v) {
    int f=fa[v],gr=fa[f];
    int sn=v==ch[f][1],son=ch[v][!sn];
    if(!isroot(f)) ch[gr][f==ch[gr][1]]=v;
    ch[f][sn]=son,ch[v][!sn]=f;
    fa[v]=gr,fa[f]=v;
    if(son) fa[son]=f;
    update(f),update(v);
}

void splay(int v) {
    while(!isroot(v)) {
        int f=fa[v],gr=fa[f];
        if(!isroot(f)) rot(f==ch[gr][1]^v==ch[f][1]?f:v);
        rot(v);
    }
}

void access(int v) {
    int tem=0;
    while(v) {
        splay(v);
        rs=tem;
        update(v);
        tem=v,v=fa[v];
    }
}

int Find_root(int v) {
    access(v);
    splay(v);
    while(ls) v=ls;
    return v;
}

void Link(int v,int f) {
    splay(v);
    fa[v]=f;
}

void Cut(int v) {
    access(v),splay(v);
    fa[ls]=0;
    ls=0;
    update(v);
}

void dfs(int v) {
    if(!v) return ;
    dfs(ls),dfs(rs);
}

void work(int v) {
    access(p[v]);
    splay(p[v]);
    int na=A[p[v]]%mod;
    int nb=B[p[v]]%mod;
    na=(1-na+mod)%mod;
    if(!na) {
        if(!nb) val[v]=-2;
        else val[v]=-1;
    } else val[v]=inv[na]*nb%mod;
}

int query(int v) {
    int rt=Find_root(v);
    access(v),splay(v);
    if(val[rt]<0) return val[rt];
    else return (val[rt]*A[v]+B[v])%mod;
}

void modify(int v) {
    int top=Find_root(v);
    Cut(v);
    k[v]=Get(),p[v]=Get(),b[v]=Get();
    update(v);
    if(Find_root(v)!=Find_root(p[v])) Link(v,p[v]);
    if(Find_root(top)!=Find_root(p[top])) Link(top,p[top]);
    work(Find_root(v)),work(Find_root(top));
}

int main() {
    A[0]=1;
    inv[0]=inv[1]=1;
    for(int i=2;i<mod;i++) inv[i]=(mod-mod/i)*inv[mod%i]%mod;
    n=Get();
    for(int i=1;i<=n;i++) {
        k[i]=Get(),p[i]=Get(),b[i]=Get();
        update(i);
    }
    for(int i=1;i<=n;i++) {
        if(Find_root(i)!=Find_root(p[i])) Link(i,p[i]);
    }
    for(int i=1;i<=n;i++) if(i==Find_root(i)) work(i);
    
    int m=Get();
    char op;
    int tim=0;
    while(m--) {
        while(op=getchar(),!isalpha(op));
        int a;
        if(op=='A') {
            a=Get();
            cout<<query(a)<<"\n";
        } else {
            a=Get();
            modify(a);
        }
    }
    
    return 0;
}