bzoj3639: Query on a tree VII【LCT+set】

題目大意：

給一棵黑白帶點權樹，支持單點翻轉顏色，單點修改權值，詢問某點同色連通塊內點權最大值。
n,m<=100000

解題思路：

同Qtree6，把虛邊維護size改爲用multiset維護最大點權即可。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int getint()
{
    int i=0,f=1;char c;
    for(c=getchar();(c!='-')&&(c<'0'||c>'9');c=getchar());
    if(c=='-')c=getchar(),f=-1;
    for(;c>='0'&&c<='9';c=getchar())i=(i<<3)+(i<<1)+c-'0';
    return i*f;
}
const int N=100005,INF=0x3f3f3f3f;
int n,m,ans,c[N],fa[N];
int tot,first[N],nxt[N<<1],to[N<<1];
struct node
{
    int fa[N],son[N][2],a[N],mx[N];
    multiset<int>s[N];
    inline int which(int x){return son[fa[x]][1]==x;}
    inline bool rt(int x){return son[fa[x]][0]!=x&&son[fa[x]][1]!=x;}
    inline void update(int x)
    {
        mx[x]=max(a[x],max(mx[son[x][0]],mx[son[x][1]]));
        if(s[x].size())mx[x]=max(mx[x],*(--s[x].end()));
    }
    inline void rotate(int x)
    {
        int y=fa[x],z=fa[y],t=which(x);
        if(!rt(y))son[z][which(y)]=x;
        fa[x]=z,fa[y]=x;
        son[y][t]=son[x][t^1],son[x][t^1]=y;
        if(son[y][t])fa[son[y][t]]=y;
        update(y),update(x);
    }
    inline void splay(int x)
    {
        while(!rt(x))
        {
            if(!rt(fa[x]))rotate(which(x)==which(fa[x])?fa[x]:x);
            rotate(x);
        }
    }
    inline void access(int x)
    {
        for(int y=0;x;y=x,x=fa[x])
        {
            splay(x);
            if(y)s[x].erase(s[x].find(mx[y]));
            if(son[x][1])s[x].insert(mx[son[x][1]]);
            son[x][1]=y,update(x);
        }
    }
    inline int findroot(int x)
    {
        access(x),splay(x);
        while(son[x][0])x=son[x][0];
        return x;
    }
    inline void link(int x,int y)
    {
        access(y),splay(y),splay(x);
        fa[x]=y,s[y].insert(mx[x]),update(y);
    }
    inline void cut(int x,int y)
    {
        access(x),splay(x);
        fa[son[x][0]]=0,son[x][0]=0,update(x);
    }
    inline void modify(int x,int y)
    {
        access(x),splay(x),a[x]=y,update(x);
    }
    inline int query(int x)
    {
        int y=findroot(x);splay(y);
        return c[x]==c[y]?mx[y]:mx[son[y][1]];
    }
}T[2];
void add(int x,int y)
{
    nxt[++tot]=first[x],first[x]=tot,to[tot]=y;
}
void dfs(int u)
{
    if(fa[u])T[c[u]].link(u,fa[u]);
    for(int e=first[u];e;e=nxt[e])
    {
        int v=to[e];if(v==fa[u])continue;
        fa[v]=u,dfs(v);
    }
}
int main()
{
    //freopen("lx.in","r",stdin);
    //freopen("lx.out","w",stdout);
    n=getint();
    for(int i=1;i<n;i++)
    {
        int x=getint(),y=getint();
        add(x,y),add(y,x);
    }
    for(int i=1;i<=n;i++)c[i]=getint();
    for(int i=1;i<=n;i++)T[0].a[i]=T[1].a[i]=T[0].mx[i]=T[1].mx[i]=getint();
    T[0].mx[0]=T[1].mx[0]=-INF;dfs(1);
    m=getint();
    while(m--)
    {
        int op=getint(),x=getint(),y;
        if(!op)printf("%d\n",T[c[x]].query(x));
        else if(op==1)
        {
            if(fa[x])T[c[x]].cut(x,fa[x]),T[c[x]^1].link(x,fa[x]);
            c[x]^=1;
        }
        else y=getint(),T[0].modify(x,y),T[1].modify(x,y);
    }
    return 0;
}