bzoj千题计划243：bzoj2325: [ZJOI2011]道馆之战

#include<cstdio>
#include<iostream>
#include<algorithm>

#define N 30001

using namespace std;

int n;

int front[N],nxt[N<<1],to[N<<1],tot;

int fa[N],dep[N],siz[N];
int bl[N];

int id[N],dy[N],cnt;

bool a[N][2];

#define max(x,y) ((x)>(y) ? (x) : (y))

struct node
{
    int sum[2][2];
    int mx[2][2];
    
    node()
    {
        for(int i=0;i<2;++i)
            for(int j=0;j<2;++j)
                sum[i][j]=mx[i][j]=0;
    } 

    node operator + (node p) const
    {
        node k;
        for(int i=0;i<=1;++i)
            for(int j=0;j<=1;++j)
            {
                if(sum[i][0] && p.sum[0][j]) 
                k.sum[i][j]=max(k.sum[i][j],sum[i][0]+p.sum[0][j]);
                if(sum[i][1] && p.sum[1][j]) 
                k.sum[i][j]=max(k.sum[i][j],sum[i][1]+p.sum[1][j]); 
            }
        k.mx[0][0]=max(k.sum[0][0],k.sum[0][1]);
        k.mx[0][1]=max(k.sum[1][0],k.sum[1][1]);
        k.mx[1][0]=max(k.sum[0][0],k.sum[1][0]);
        k.mx[1][1]=max(k.sum[0][1],k.sum[1][1]);
        if(sum[0][0]) k.mx[0][0]=max(k.mx[0][0],sum[0][0]+p.mx[0][0]);
        if(sum[0][1]) k.mx[0][0]=max(k.mx[0][0],sum[0][1]+p.mx[0][1]);
        if(sum[1][0]) k.mx[0][1]=max(k.mx[0][1],sum[1][0]+p.mx[0][0]);
        if(sum[1][1]) k.mx[0][1]=max(k.mx[0][1],sum[1][1]+p.mx[0][1]);
        if(p.sum[0][0]) k.mx[1][0]=max(k.mx[1][0],mx[1][0]+p.sum[0][0]);
        if(p.sum[1][0]) k.mx[1][0]=max(k.mx[1][0],mx[1][1]+p.sum[1][0]);
        if(p.sum[0][1]) k.mx[1][1]=max(k.mx[1][1],mx[1][0]+p.sum[0][1]);
        if(p.sum[1][1]) k.mx[1][1]=max(k.mx[1][1],mx[1][1]+p.sum[1][1]);
        k.mx[0][0]=max(k.mx[0][0],mx[0][0]);
        k.mx[0][1]=max(k.mx[0][1],mx[0][1]);
        k.mx[1][0]=max(k.mx[1][0],p.mx[1][0]);
        k.mx[1][1]=max(k.mx[1][1],p.mx[1][1]); 
        return k;
    }
    
    void turn()
    {
        swap(sum[0][1],sum[1][0]);
        swap(mx[0][0],mx[1][0]);
        swap(mx[0][1],mx[1][1]);
    }

}tr[N<<2];

void read(int &x)
{
    x=0; char c=getchar();
    while(!isdigit(c)) c=getchar();
    while(isdigit(c)) { x=x*10+c-‘0‘; c=getchar(); }
}

void add(int u,int v)
{
    to[++tot]=v; nxt[tot]=front[u]; front[u]=tot;
    to[++tot]=u; nxt[tot]=front[v]; front[v]=tot;
}

void dfs1(int x)
{
    siz[x]=1;
    for(int i=front[x];i;i=nxt[i])
        if(to[i]!=fa[x])
        {
            fa[to[i]]=x;
            dep[to[i]]=dep[x]+1;
            dfs1(to[i]);
            siz[x]+=siz[to[i]];
        }
}

void dfs2(int x,int top)
{
    bl[x]=top;
    id[x]=++cnt;
    dy[cnt]=x;
    int y=0;
    for(int i=front[x];i;i=nxt[i])
        if(to[i]!=fa[x] && siz[to[i]]>siz[y]) y=to[i];
    if(y) dfs2(y,top);
    else return;
    for(int i=front[x];i;i=nxt[i])
        if(to[i]!=y && to[i]!=fa[x]) dfs2(to[i],to[i]);
}

void build(int k,int l,int r)
{
    if(l==r)
    {
        if(a[dy[l]][0]) tr[k].sum[0][0]=1;
        if(a[dy[l]][1]) tr[k].sum[1][1]=1;
        if(a[dy[l]][0] && a[dy[l]][1]) tr[k].sum[0][1]=tr[k].sum[1][0]=2;
        tr[k].mx[0][0]=max(tr[k].sum[0][0],tr[k].sum[0][1]);
        tr[k].mx[0][1]=max(tr[k].sum[1][0],tr[k].sum[1][1]);
        tr[k].mx[1][0]=max(tr[k].sum[0][0],tr[k].sum[1][0]);
        tr[k].mx[1][1]=max(tr[k].sum[0][1],tr[k].sum[1][1]);
        return;
    }
    int mid=l+r>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    tr[k]=tr[k<<1]+tr[k<<1|1];
}

node query(int k,int l,int r,int opl,int opr)
{
    if(l>=opl && r<=opr) return tr[k];
    int mid=l+r>>1;
    if(opr<=mid) return query(k<<1,l,mid,opl,opr);
    else if(opl>mid) return query(k<<1|1,mid+1,r,opl,opr);
    else return query(k<<1,l,mid,opl,opr)+query(k<<1|1,mid+1,r,opl,opr);
}

void Query(int u,int v)
{
    node ansu,ansv; 
    bool firstu=false,firstv=false;
    while(bl[u]!=bl[v])
    {
        if(dep[bl[u]]>dep[bl[v]])
        {
            if(!firstu) firstu=true,ansu=query(1,1,n,id[bl[u]],id[u]);
            else ansu=query(1,1,n,id[bl[u]],id[u])+ansu;
            u=fa[bl[u]];
        }
        else
        {
            if(!firstv) firstv=true,ansv=query(1,1,n,id[bl[v]],id[v]);
            else ansv=query(1,1,n,id[bl[v]],id[v])+ansv;
            v=fa[bl[v]];
        }
    }
    if(dep[u]>dep[v])
    {
        if(!firstu) firstu=true,ansu=query(1,1,n,id[v],id[u]);
        else ansu=query(1,1,n,id[v],id[u])+ansu;
    }
    else
    {
        if(!firstv) firstv=true,ansv=query(1,1,n,id[u],id[v]);
        else ansv=query(1,1,n,id[u],id[v])+ansv;
    }
    if(!firstu) ansu=ansv;
    else
    {
        ansu.turn();
        if(firstv) ansu=ansu+ansv;
    }
    cout<<max(ansu.mx[0][0],ansu.mx[0][1])<<‘\n‘;
}

void change(int k,int l,int r,int x,bool u,bool v)
{
    if(l==r)
    {
        a[l][0]=u;
        a[l][1]=v;
        if(a[l][0]) tr[k].sum[0][0]=1;
        else tr[k].sum[0][0]=0;
        if(a[l][1]) tr[k].sum[1][1]=1;
        else tr[k].sum[1][1]=0;
        if(a[l][0] && a[l][1]) tr[k].sum[0][1]=tr[k].sum[1][0]=2;
        else tr[k].sum[0][1]=tr[k].sum[1][0]=0;
        tr[k].mx[0][0]=max(tr[k].sum[0][0],tr[k].sum[0][1]);
        tr[k].mx[0][1]=max(tr[k].sum[1][0],tr[k].sum[1][1]);
        tr[k].mx[1][0]=max(tr[k].sum[0][0],tr[k].sum[1][0]);
        tr[k].mx[1][1]=max(tr[k].sum[0][1],tr[k].sum[1][1]);
        return;
    }
    int mid=l+r>>1;
    if(x<=mid) change(k<<1,l,mid,x,u,v);
    else change(k<<1|1,mid+1,r,x,u,v);
    tr[k]=tr[k<<1]+tr[k<<1|1];
}

int main()
{
    freopen("fight.in","r",stdin);
    freopen("fight.out","w",stdout); 
    int m;
    read(n); read(m);
    int u,v;
    for(int i=1;i<n;++i)
    {
        read(u); read(v);
        add(u,v);
    }
    dfs1(1);
    dfs2(1,1);
    char s[3];
    for(int i=1;i<=n;++i)
    {
        scanf("%s",s);
        if(s[0]==‘.‘) a[i][0]=true;
        if(s[1]==‘.‘) a[i][1]=true;
    }
    build(1,1,n);
    char c[3];
    while(m--)
    {
        scanf("%s",c);
        if(c[0]==‘Q‘) 
        {
            read(u); read(v);
            Query(u,v);
        }
        else
        {
            read(u);
            scanf("%s",c);
            change(1,1,n,id[u],c[0]==‘.‘,c[1]==‘.‘);
        }
    }
}