[Luogu P3950] 部落冲突

COGS
Luogu
另外值得一提的是 COGS上有这么多点

题意描述：

给你有N个点的一棵树。
C u v 表示 u v之间的道路不能通行
Q u v 表示 询问 u v之间是否有不可通行的道路
U u 表示 第u个 C操作失效
写一种数据结构，支持以上三种操作。

题目分析：

树剖解决！
我们先进行树剖，然后维护一颗边权线段树，支持区间修改和区间查询。
每次把封锁道路的时间视为 u->lca(u,v)->v上的路权值+1
查询的时候只要查询 u->lca(u,v)->v这条路上的权值是否为0即可

AC Code

#include <cstdio>
#include <iostream>
#define lson (o<<1),l,mid
#define rson (o<<1)+1,mid+1,r
using namespace std;
const int maxm=310000;
int deep[maxm],son[maxm],size[maxm],fa[maxm],top[maxm];
int head[maxm],to[maxm<<1],net[maxm<<1],cnt;
int p[maxm],val[maxm];
int id[maxm],tot;
int n,m,root;
struct node{
    int l,r,siz,maxi,sum,add;
};
node t[maxm*4];
struct war{
    int u,v;
};
war a[maxm];
inline void add(int x,int y)
{
    to[++cnt]=y;
    net[cnt]=head[x];
    head[x]=cnt;
}
int dfs1(int now,int fax,int dep)
{
    fa[now]=fax,deep[now]=dep,size[now]=1;
    int maxson=-1;
    for(int i=head[now];i;i=net[i])
    if(to[i]!=fax)
    {
        size[now]+=dfs1(to[i],now,dep+1);
        if(maxson<size[to[i]])
         son[now]=to[i],maxson=size[to[i]];
    }
    return size[now];
}
void dfs2(int now,int topx)
{
    id[now]=++tot;
    val[tot]=p[now];
    top[now]=topx;
    if(!son[now]) return;
    dfs2(son[now],topx);
    for(int i=head[now];i;i=net[i])
     if(!id[to[i]])
      dfs2(to[i],to[i]);
}
inline void update(int o)
{
    t[o].sum=(t[(o<<1)].sum+t[(o<<1)|1].sum);
    t[o].maxi=max(t[(o<<1)].maxi,t[(o<<1)|1].maxi);
}
inline void pushdown(int o)
{
    int adi=t[o].add;
    for(int i=0;i<=1;i++)
     t[(o<<1)+i].sum=(t[(o<<1)+i].sum+(t[(o<<1)+i].siz*adi)),t[(o<<1)+i].add=(t[(o<<1)+i].add+adi);
    t[o].add=0;
}
void build(int o,int l,int r)
{
    t[o].l=l,t[o].r=r,t[o].siz=r-l+1;
    if(l==r)
    {
        t[o].sum=t[o].maxi=val[l];
        return;
    }
    int mid=(l+r)>>1;
    build(lson),build(rson);
    update(o);
}
void adj(int o,int ql,int qr,int num)
{
    int l=t[o].l,r=t[o].r;
    if(ql<=l&&qr>=r)
    {
        t[o].sum=(t[o].sum+(t[o].siz*num));
        t[o].add=(t[o].add+num);
        return;
    }
    pushdown(o);
    int mid=(l+r)>>1;
    if(ql<=mid) adj((o<<1),ql,qr,num);
    if(qr>mid) adj((o<<1)|1,ql,qr,num);
    update(o);
}
int ask(int o,int ql,int qr)
{
    int l=t[o].l,r=t[o].r;
    if(ql<=l&&qr>=r)
     return t[o].sum;
    pushdown(o);
    int mid=(l+r)>>1;
    int p1=0,p2=0;
    if(ql<=mid) p1=ask((o<<1),ql,qr);
    if(qr>mid) p2=ask((o<<1)|1,ql,qr);
    return (p1+p2);
}
inline int tree_sum(int u,int v)
{
    int ans=0;
    while(top[u]!=top[v])
    {
        if(deep[top[u]]<deep[top[v]]) swap(u,v);
        ans=(ans+ask(1,id[top[u]],id[u]));
        if(ans) return ans;
        u=fa[top[u]];
    }
    if(deep[u]<deep[v]) swap(u,v);
    ans=(ans+ask(1,id[v]+1,id[u]));
    return ans;
}
inline void tree_x(int u,int v,int num)
{
    while(top[u]!=top[v])
    {
        if(deep[top[u]]<deep[top[v]]) swap(u,v);
        adj(1,id[top[u]],id[u],num);
        u=fa[top[u]];
    }
    if(deep[u]<deep[v]) swap(u,v);
    adj(1,id[v]+1,id[u],num);
}
inline int read()
{
    int x=0,w=1;char ch=0;
    while(ch<'0'||ch>'9') 
    {
        if(ch=='-') w=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*w;
}
int main()
{   
    freopen("lct.in","r",stdin);
    freopen("lct.out","w",stdout);
    n=read(),m=read();
    for(int i=1;i<n;i++)
    {
        int u=read(),v=read();
        add(u,v),add(v,u);
    }
    dfs1(1,0,0),dfs2(1,1);
    build(1,1,tot);
    char s[10];
    int nows=0;
    for(int i=1,u,v,x;i<=m;i++)
    {
        scanf("%s",s);
        if(s[0]=='U')
         x=read(),tree_x(a[x].u,a[x].v,-1);
        if(s[0]=='Q')
        {
            u=read(),v=read();
            if(tree_sum(u,v)) printf("No\n");
            else printf("Yes\n");
        }
        if(s[0]=='C')
        {
           a[++nows].u=read(),a[nows].v=read();
           tree_x(a[nows].u,a[nows].v,1);
        }
    }
    return 0;
}