CF343D Water Tree（线段树+dfs序+思路）

首先对于操作一，我们按照老套路把树上结点用dfs序变成序列，每个结点的子树均为in[x]~out[x]连续的一段，段修改即可。对于操作2，往上找父亲肯定不好实现，怎么办呢？我们只改这一个点，把他改为0，然后在询问x点的时候，我们查询x的子树中是否有0，有的话他一定也是0（他一定是那个点的父亲）。这样就很巧妙的把一个点会影响他的所有祖先转化成了单点修改，查询子树。需要注意的是，这样做的话，做操作1时要检查一下x的子树是否有0，如果有的话，我们不仅要把x的子树都变成1，还要把0这个标记传递给x的父亲，他们还是没有水的。

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 500010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,h[N],num=0,in[N],out[N],dfn=0,fa[N];
struct edge{
    int to,next;
}data[N<<1];
void dfs(int x){
    in[x]=++dfn;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(fa[x]==y) continue;
        fa[y]=x;dfs(y);
    }out[x]=dfn;
}
struct node{
    bool x;
}tree[N<<2];
inline void pushup(int p){
    tree[p].x=tree[p<<1].x&tree[p<<1|1].x;
}
inline void pushdown(int p){
    if(tree[p].x) tree[p<<1].x=tree[p<<1|1].x=1;
}
void build(int p,int l,int r){
    if(l==r) return;
    int mid=l+r>>1;
    build(p<<1,l,mid);build(p<<1|1,mid+1,r);
}
void update(int p,int l,int r,int x,int y){
    if(x<=l&&r<=y){tree[p].x=1;return;}
    int mid=l+r>>1;
    if(x<=mid) update(p<<1,l,mid,x,y);
    if(y>mid) update(p<<1|1,mid+1,r,x,y);
    pushup(p);
}
void change(int p,int l,int r,int x){
    if(l==r){tree[p].x=0;return;}
    int mid=l+r>>1;pushdown(p);
    if(x<=mid) change(p<<1,l,mid,x);
    else change(p<<1|1,mid+1,r,x);
    pushup(p);
}
int ask(int p,int l,int r,int x,int y){
    if(x<=l&&r<=y) return tree[p].x;
    int mid=l+r>>1,res=1;pushdown(p);
    if(x<=mid) res&=ask(p<<1,l,mid,x,y);
    if(y>mid) res&=ask(p<<1|1,mid+1,r,x,y);
    return res;
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }dfs(1);m=read();build(1,1,n);
    while(m--){
        int op=read(),x=read();
        if(op==1){
            if(x!=1&&ask(1,1,n,in[x],out[x])==0) change(1,1,n,in[fa[x]]);
            update(1,1,n,in[x],out[x]);
        }
        if(op==2) change(1,1,n,in[x]);
        if(op==3) printf("%d\n",ask(1,1,n,in[x],out[x]));
    }
    return 0;
}