P3950 部落冲突（树链剖分）

题目描述

简单起见，你就是要处理下面三件事，所有的事件都是按照时间顺序给出的。

1. Q p q 从第 pp 个部落出发的建筑工人想知道能否到达第 qq 个部落了，你要回答的便是 Yes / No，注意大小写。

2. C p q 第 pp 个部落与第 qq 个部落开战了，保证他们一定是相邻的部落，且目前处于停战（未开战）状态。

3. U x 第 xx 次发生的战争结束了，它将永远的被载入史册，不复存在（保证这个消息不会告诉你多次）

输入格式

第一行两个数 nn 和 mm， nn 代表了一共有 nn 个部落，mm 代表了以上三种事件发生的总数。

接下来的 n - 1n−1 行，每行两个数 p, qp,q，代表了第 pp 个部落与第 qq 个部落之间有一条道路相连。

接下来的 mm 行，每行表示一件事，详见题目描述。

输出格式

每行一个 Yes 或者 No，表示从第 pp 个部落出发的建筑工人能否到达第 qq 个部落。

#include<bits/stdc++.h>
using namespace std;
const int maxn=3e5+100;
vector<int> g[maxn];
struct war {
    int u,v;
}War[maxn];
int n,m;
int son[maxn];
int id[maxn];
int fa[maxn];
int cnt;
int dep[maxn];
int size[maxn];
int top[maxn];
int w[maxn];
int wt[maxn];

struct node {
    int l,r;
    int sum;
    int lazy;
}segTree[maxn*4];
void build (int i,int l,int r) {
    segTree[i].l=l;
    segTree[i].r=r;
    if (l==r) {
        segTree[i].sum=wt[l];
        return;
    }
    int mid=(l+r)>>1;
    build(i<<1,l,mid);
    build(i<<1|1,mid+1,r);
    segTree[i].sum=segTree[i<<1].sum+segTree[i<<1|1].sum; 
}
void spread (int i) {
    if (segTree[i].lazy) {
        segTree[i<<1].sum+=segTree[i].lazy*(segTree[i<<1].r-segTree[i<<1].l+1);
        segTree[i<<1|1].sum+=segTree[i].lazy*(segTree[i<<1|1].r-segTree[i<<1|1].l+1);
        segTree[i<<1].lazy+=segTree[i].lazy;
        segTree[i<<1|1].lazy+=segTree[i].lazy;
        segTree[i].lazy=0;
    }
}
void update (int i,int l,int r,int val) {
    if (l<=segTree[i].l&&segTree[i].r<=r) {
        segTree[i].sum+=val*(segTree[i].r-segTree[i].l+1);
        segTree[i].lazy+=val;
        return;
    }
    spread(i);
    int mid=(segTree[i].l+segTree[i].r)>>1;
    if (l<=mid)
        update(i<<1,l,r,val);
    if (r>mid)
        update(i<<1|1,l,r,val);
    segTree[i].sum=segTree[i<<1].sum+segTree[i<<1|1].sum;
}
int query (int i,int l,int r) {
    if (l<=segTree[i].l&&segTree[i].r<=r) 
        return segTree[i].sum;
    spread(i);
    int mid=(segTree[i].l+segTree[i].r)>>1;
    int ans=0;
    if (l<=mid)
        ans+=query(i<<1,l,r);
    if (r>mid)
        ans+=query(i<<1|1,l,r);
    return ans;
}
int qRange (int x,int y) {
    int ans=0;
    while (top[x]!=top[y]) {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        ans+=query(1,id[top[x]],id[x]);
        if (ans) return ans; 
        x=fa[top[x]];
    }
    if (dep[x]>dep[y]) swap(x,y);
    ans+=query(1,id[x]+1,id[y]);
    return ans;
}
void upRange (int x,int y,int k) {
    //修改整条路径 
    while (top[x]!=top[y]) {
        if (dep[top[x]]<dep[top[y]]) swap(x,y);
        update(1,id[top[x]],id[x],k);
        x=fa[top[x]];
    }
    if (dep[x]>dep[y]) swap(x,y);
    update(1,id[x]+1,id[y],k);
}
void dfs1 (int x,int f,int deep) {
    dep[x]=deep;
    fa[x]=f;
    size[x]=1;
    int maxson=-1;//记录重儿子的儿子数
    for (int y:g[x]) {
        if (y==f) continue;
        dfs1(y,x,deep+1);
        size[x]+=size[y];
        if (size[y]>maxson) son[x]=y,maxson=size[y];
    } 
}
void dfs2 (int x,int topf) {
    id[x]=++cnt;
    wt[cnt]=w[x];
    top[x]=topf;
    if (!son[x]) return;
    dfs2(son[x],topf);
    for (int y:g[x]) {
        if (y==fa[x]||y==son[x]) continue;
        dfs2(y,y);
    }
}
int main () {
    scanf("%d%d",&n,&m);
    for (int i=1;i<n;i++) {
        int x,y;
        scanf("%d%d",&x,&y);
        g[x].push_back(y);
        g[y].push_back(x);
    }
    dfs1(1,0,1);
    dfs2(1,1);
    build(1,1,n);
    int tot=0;
    for (int i=1;i<=m;i++) {
        string s;
        cin>>s;
        if (s=="Q") {
            int x,y;
            scanf("%d%d",&x,&y);
            int ans=qRange(x,y);
            if (ans)
                printf("No\n");
            else
                printf("Yes\n");
        }
        else if (s=="C") {
            int x,y;
            scanf("%d%d",&x,&y);
            War[++tot]={x,y};
            upRange(x,y,1);
        }
        else if (s=="U") {
            int x;
            scanf("%d",&x);
            upRange(War[x].u,War[x].v,-1);
        }
    }
}