WISCO信息组NOIP模拟赛-部落冲突

传送门

---

 首先肯定考虑树剖，这里没有要求区间加，所以可以用树状数组维护，不会卡常的

这里是边权，可以转化为点权：让每条边连接的较深的节点的点权等于边权即可，然后计算的时候减去lca

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#define MAXN 300005
#define LOG 20
using namespace std;
int read(){
    int x=0;char ch=getchar();
    while(ch<'0'||ch>'9'){ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+(ch^48);ch=getchar();}
    return x;
}
int n,T;
int a[MAXN],dat[MAXN];
int dep[MAXN],size[MAXN],gs[MAXN],fa[20][MAXN];
int top[MAXN],tree[MAXN],pre[MAXN],tot;
int first[MAXN],nxt[MAXN<<1],to[MAXN<<1],cnt;
int id[MAXN],tmp;

void add(int x,int y){
    nxt[++cnt]=first[x];first[x]=cnt;to[cnt]=y;
    nxt[++cnt]=first[y];first[y]=cnt;to[cnt]=x;
}
int lca(int x,int y){
    if(dep[x]<dep[y]){
        swap(x,y);
    }
    for(int k=dep[x]-dep[y],p=0;k;k>>=1,p++){
        if(k&1){
            x=fa[p][x];
        }
    }
    if(x==y){
        return x;
    }
    for(int k=LOG-1;k>=0;k--){
        if(fa[k][x]!=fa[k][y]){
            x=fa[k][x],y=fa[k][y];
        }
    }
    return fa[0][x];
}
void change(int k,int x){
    while(k<=n){
        dat[k]+=x;
        k+=(k&-k);
    }
}
int query(int k){
    int ret=0;
    while(k>=1){
        ret+=dat[k];
        k-=(k&-k);
    }
    return ret;
}
void dfs1(int x){
    size[x]=1;
    for(int e=first[x];e;e=nxt[e]){
        int y=to[e];
        if(y==fa[0][x]){
            continue;
        }
        fa[0][y]=x;
        dep[y]=dep[x]+1;
        dfs1(y);
        size[x]+=size[y];
        if(size[y]>size[gs[x]]){
            gs[x]=y;
        }
    }
}
void dfs2(int x,int t){
    top[x]=t;
    tree[x]=(++tot);
    pre[tot]=x;
    if(!gs[x]){
        return;
    }
    dfs2(gs[x],t);
    for(int e=first[x];e;e=nxt[e]){
        int y=to[e];
        if(y==fa[0][x]||y==gs[x]){
            continue;
        }
        dfs2(y,y);
    }
}
int ask(int x,int y){
    int f1=top[x],f2=top[y];
    if(dep[f1]<dep[f2]){
        swap(x,y),swap(f1,f2);
    }
    int ret=-a[lca(x,y)];
    while(f1!=f2){
        ret+=query(tree[x])-query(tree[f1]-1);
        x=fa[0][f1]; f1=top[x];
        if(dep[f1]<dep[f2]){
            swap(x,y),swap(f1,f2);
        }    
    }
    if(dep[x]<dep[y]){
        swap(x,y);
    }
    ret+=query(tree[x])-query(tree[y]-1);
    return (ret<=0);
}
void init(){
    n=read();T=read();
    for(int i=1;i<n;i++){
        int x=read(),y=read();
        add(x,y);
    }
    dfs1(1);
    dfs2(1,1);
    for(int k=1;k<LOG;k++){
        for(int i=1;i<=n;i++){
            fa[k][i]=fa[k-1][fa[k-1][i]];
        }
    }
}
void solve(){
    char ch[5];
    while(T--){
        scanf("%s",ch);
        int x=read();
        if('Q'==ch[0]){
            int y=read();
            if(ask(x,y)){
                printf("Yes\n");
            }
            else{
                printf("No\n");
            }
        }
        else if('C'==ch[0]){
            int y=read();
            if(dep[x]<dep[y]){
                swap(x,y);
            }
            change(tree[x],1);
            a[x]++;
            id[++tmp]=x;
        }
        else{
            x=id[x];
            change(tree[x],-1);
            a[x]--;
        }
    }
}
int main()
{
    init();
    solve();
    return 0;
}

也可以是树上差分，用树状数组+dfs序，本质上是差不多的

#include<cstdio>
#include<cstdlib>
#include<algorithm>
#include<cstring>
#define MAXN 300005
#define LOG 20
using namespace std;
int read(){
    int x=0;char ch=getchar();
    while(ch<'0'||ch>'9'){ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+(ch^48);ch=getchar();}
    return x;
}
int n,T;
int first[MAXN],nxt[MAXN<<1],to[MAXN<<1],cnt;
int pin[MAXN],pout[MAXN],tot;
int dat[MAXN<<1];
int fa[LOG][MAXN],dep[MAXN];
int id[MAXN],tmp;
void change(int k,int x){
    while(k<=(n<<1)){
        dat[k]+=x;
        k+=(k&-k);
    }
}
int query(int k){
    int ret=0;
    while(k>=1){
        ret+=dat[k];
        k-=(k&-k);
    }
    return ret;
}
void add(int x,int y){
    nxt[++cnt]=first[x];first[x]=cnt;to[cnt]=y;
    nxt[++cnt]=first[y];first[y]=cnt;to[cnt]=x;
}
int lca(int x,int y){
    if(dep[x]<dep[y]){
        swap(x,y);
    }
    for(int k=dep[x]-dep[y],p=0;k;k>>=1,p++){
        if(k&1){
            x=fa[p][x];
        }
    }
    if(x==y){
        return x;
    }
    for(int k=LOG-1;k>=0;k--){
        if(fa[k][x]!=fa[k][y]){
            x=fa[k][x],y=fa[k][y];
        }
    }
    return fa[0][x];
}
void dfs(int x){
    pin[x]=(++tot);
    for(int e=first[x];e;e=nxt[e]){
        int y=to[e];
        if(y==fa[0][x]){
            continue;
        }
        dep[y]=dep[x]+1;
        fa[0][y]=x;
        dfs(y);
    }
    pout[x]=(++tot);
}
int ask(int x,int y){
    int t=query(pin[x])+query(pin[y])-2*query(pin[lca(x,y)]);
    return (t<=0);
}
void init(){
    n=read(); T=read();
    for(int i=1;i<n;i++){
        int x=read(),y=read();
        add(x,y);
    }    
    dfs(1);
    for(int k=1;k<LOG;k++){
        for(int i=1;i<=n;i++){
            fa[k][i]=fa[k-1][fa[k-1][i]];
        }
    }
}
void solve(){
    char ch[5]={0};
    while(T--){
        scanf("%s",ch);
        int x=read();
        if('Q'==ch[0]){
            int y=read();
            if(ask(x,y)){
                printf("Yes\n");
            }
            else{
                printf("No\n");
            }
        }
        else if('C'==ch[0]){
            int y=read();
            if(dep[x]<dep[y]){
                swap(x,y);
            }
            change(pin[x],1);
            change(pout[x]+1,-1);
            id[++tmp]=x;
        }
        else{
            x=id[x];
            change(pin[x],-1);
            change(pout[x]+1,1);
        }
    }
}
int main()
{
//    freopen("data.in","r",stdin);
    init();
    solve();
    return 0;
}