D. Water Tree

模板题

#include<iostream>
#include<vector>
using namespace std;
const int N=5e5+9;
int n;
//树剖
//1.转成线性部分
vector<int> e[N];
void add(int u,int v){
    e[u].push_back(v);
    e[v].push_back(u);
}
int fa[N],dep[N],sz[N],wc[N];
void dfs1(int u,int f){//fa dep sz wc
    fa[u]=f;
    sz[u]=1;
    dep[u]=dep[f]+1;
    for(auto & i : e[u]){
        if(i!=f){
            dfs1(i,u);
            sz[u]+=sz[i];
            if(sz[i]>sz[wc[u]]){
                wc[u]=i;
            }
        }
    }
}
int dfn[N],rdfn[N],top[N],vistime;
void dfs2(int u,int Top){//dfn rdfn top
    dfn[u]=++vistime;
    rdfn[vistime]=u;
    top[u]=Top;
    if(wc[u]){
        dfs2(wc[u],Top);
        for(auto & i : e[u]){
            if(i!=wc[u] && i!=fa[u]){
                dfs2(i,i);
            }
        }
    }
}
//2.线段树维护
struct SEG{
    #define INF (1<<31)
    #define ll long long
    #define tl(id) (id<<1)
    #define tr(id) (id<<1|1)
    #define li inline
    struct node{
        int val,tag;
    }seg[N<<2];
    #define pushup(id) seg[id].val=seg[tl(id)].val+seg[tr(id)].val
    li int inrange(int L,int R,int l,int r){return l<=L && R<=r;}
    li int outofrange(int L,int R,int l,int r){return L>r || R<l;}
    li void build(const int id,int l,int r){
    	seg[id].tag=INF;
        if(l==r){
            seg[id].val=0;
            return;
        }
        int mid=(l+r)>>1;
        build(tl(id),l,mid);
        build(tr(id),mid+1,r);
        pushup(id);
    }
    li void maketag(int id,int l,int r,ll v){
        seg[id].val=(r-l+1)*v;
        seg[id].tag=v;
    }
    li void pushdown(int id,int l,int r){
     	if(seg[id].tag==INF){
     		return;
     	}
        int mid=(l+r)>>1;
        maketag(tl(id),l,mid,seg[id].tag);
        maketag(tr(id),mid+1,r,seg[id].tag);
        seg[id].tag=INF;
    }
    li ll query(int id,int L,int R,int l,int r){
        if(inrange(L,R,l,r)){
            return seg[id].val;
        }else if(!outofrange(L,R,l,r)){
            int mid=(L+R)>>1;
            pushdown(id,L,R);
            return (query(tl(id),L,mid,l,r)+query(tr(id),mid+1,R,l,r));
        }else{
            return 0;
        }
    }
    li void modify(int id,int L,int R,int l,int r,ll x){
        if(inrange(L,R,l,r)){
            maketag(id,L,R,x);
        }else if(!outofrange(L,R,l,r)){
            int mid=(L+R)>>1;
            pushdown(id,L,R);
            modify(tl(id),L,mid,l,r,x);
            modify(tr(id),mid+1,R,l,r,x);
            pushup(id);
        }
    }
}t;
//3.找LCA，同时完成操作
void update(int u,int v,int z){
    while(top[u]!=top[v]){
        if(dep[top[u]]<dep[top[v]]){//
            swap(u,v);
        }
        t.modify(1,1,n,dfn[top[u]],dfn[u],z);//深度越小，dfs序号越小
        u=fa[top[u]];//处理完上移动
    }
    t.modify(1,1,n,min(dfn[u],dfn[v]),max(dfn[u],dfn[v]),z);
}
ll ask(int u,int v){
    ll res=0;
    while(top[u]!=top[v]){
        if(dep[top[u]]<dep[top[v]]){
            swap(u,v);
        }
        res+=t.query(1,1,n,dfn[top[u]],dfn[u]);
        u=fa[top[u]];
    }
    res+=t.query(1,1,n,min(dfn[u],dfn[v]),max(dfn[u],dfn[v]));
    return res;
}
int main(){
	cin>>n;
	for(int i=1;i<=n-1;i++){
		int u,v;
		cin>>u>>v;
		add(u,v);
	}
	dfs1(1,0);
	dfs2(1,1);
	t.build(1,1,n);
	int m;
	cin>>m;
	for(int i=1;i<=m;i++){
		int op;
		cin>>op;
		if(op==1){
			int u;
			cin>>u;
			t.modify(1,1,n,dfn[u],dfn[u]+sz[u]-1,1);
		}else if(op==2){
			int u;
			cin>>u;
			update(1,u,0);
		}else{
			int u;
			cin>>u;
			cout<<t.query(1,1,n,dfn[u],dfn[u])<<'\n';
		}
	}
	return 0;
}

题外话，状态不好，不要写，不然板子题都会写bug,瞪眼法看了半小时看不出来重新写直接ac了

服了