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 500010inline 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;}