BZOJ1095 捉迷藏

类似QTREE4

LCT维护子树信息即可

代码

#include
#define For(i,j,k) for(int i=(j);i<=(int)k;i++)
#define Forr(i,j,k) for(int i=(j);i>=(int)k;i--)
#define Set(a,b) memset(a,b,sizeof(a))
#define Rep(i,u) for(int i=Begin[u],v=to[i];i;i=Next[i],v=to[i])
#define L(i) (T[i].s[0])
#define R(i) (T[i].s[1])
#define F(i) (T[i].fa)
#define lmx(i) (T[i].lmx)
#define rmx(i) (T[i].rmx)
#define mx(i) (T[i].mx)
#define Loc(i) (R(F(i))==i)
#define S(i) (T[i].sum)
using namespace std;
const int N=200010 ,INF=0x3f3f3f3f;
inline void read(int &x){
    x=0;char c=getchar();int f(0);
    while(c<'0'|| c>'9')f|=(c=='-'),c=getchar();
    while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=getchar();
    if(f)x=-x;
}
int n,m,e,Begin[N],Next[N],to[N],w[N],ans;
inline void add(int x,int y,int z){
    to[++e]=y,Next[e]=Begin[x],Begin[x]=e,w[e]=z;
}
inline int fir(multiset<int> &s){return s.size()?*s.rbegin():-INF;}
inline int sec(multiset<int> &s){return s.size()>1?*(++s.rbegin()):-INF;}
struct node{
    int s[2],lmx,rmx,mx,sum,len,w,fa; 
    multiset<int>chain,path;
};
struct LCT{
    node T[N];
    inline void init(){
        For(i,0,n)lmx(i)=rmx(i)=mx(i)=-INF;
    }
    inline void pushup(int x){
        S(x)=S(L(x))+S(R(x))+T[x].len;
        int cha=max(T[x].w,fir(T[x].chain));
        int l=max(cha,rmx(L(x))+T[x].len);
        int r=max(cha,lmx(R(x)));
        lmx(x)=max(lmx(L(x)),S(L(x))+T[x].len+r);
        rmx(x)=max(rmx(R(x)),S(R(x))+l);
        mx(x)=max(mx(L(x)),mx(R(x)));
        mx(x)=max(mx(x),fir(T[x].chain)+sec(T[x].chain));
        mx(x)=max(mx(x),fir(T[x].path));
        mx(x)=max(mx(x),max(rmx(L(x))+T[x].len+r,lmx(R(x))+l));
        if(T[x].w==0)mx(x)=max(mx(x),max(fir(T[x].chain),0));
    }
    inline bool isrt(int x){
        return R(F(x))!=x&&L(F(x))!=x;
    }
    inline void Rotate(int x){
        int A=F(x),B=F(A),l=Loc(x),r=l^1,d=Loc(A);
        if(!isrt(A))T[B].s[d]=x;F(x)=B;
        F(A)=x,F(T[x].s[r])=A,T[A].s[l]=T[x].s[r],T[x].s[r]=A;
        pushup(A);
    }
    inline void splay(int x){
        while(!isrt(x)){
            int y=F(x);
            if(!isrt(y))Rotate(x);
            Rotate(x);
        }
        pushup(x);
    }
    inline void access(int x){
        for(int i=0;x;i=x,x=F(x)){
            splay(x);
            if(R(x))T[x].chain.insert(lmx(R(x))),T[x].path.insert(mx(R(x)));
            if(i)T[x].chain.erase(T[x].chain.find(lmx(i))),T[x].path.erase(T[x].path.find(mx(i)));
            R(x)=i,pushup(x);
        }
    }
    inline void modify(int x){
        access(x);splay(x),T[x].w=(T[x].w==0)?-INF:0;pushup(x);
        ans=mx(x);
    }
}t;
void dfs(int u){
    Rep(i,u)
        if(t.T[u].fa!=v)
            t.T[v].fa=u,t.T[v].len=w[i],dfs(v),
            t.T[u].chain.insert(t.T[v].lmx),t.T[u].path.insert(t.T[v].mx);
    t.pushup(u);
}
char s[10];
int main(){
    read(n);t.init();
    For(i,1,n-1){
        int u,v,w;
        read(u),read(v),w=1;
        add(u,v,w),add(v,u,w);
    }   
    dfs(1);ans=t.T[1].mx;
    read(m);
    while(m--){
        int x;
        scanf("%s",s);
        if(s[0]=='G'){
            if(ans<0)puts("-1");
            else printf("%d\n",ans);
        }else 
            read(x),t.modify(x);

    }
    return 0;
}