Codeforces Round #200 (Div. 1) D. Water Tree

题目描述：戳这里

题解：

这题是一道数据结构题。
当然，我们一看操作2，就可以直接树链剖分。
但是这样写比较长= =表示不想（不会）码，所以就搜索了一下，发现了一个比较简单的方法。
我们注意到，一个节点，只要它的子树中有一个点是没有水的，那么根据操作2，这个点肯定是没有水的。
如果是操作一，直接修改一下子树就好了。
这让我们想到了线段树。先求一下dfs序，然后直接线段树修改一下就好了。

详细见代码：

#include
#include
using namespace std;
const int maxn=500005;
int n,Q,tot,a[maxn],lnk[maxn],son[2*maxn],nxt[2*maxn],p[maxn],r[maxn],fa[maxn];
struct dyt{
    int l,r,sum,tag;
}tree[5*maxn];
void add(int x,int y){
    son[++tot]=y,nxt[tot]=lnk[x],lnk[x]=tot;
}
void dfs(int x){
    p[x]=++tot;
    for (int j=lnk[x];j;j=nxt[j])
    if (!p[son[j]]) {
        fa[son[j]]=x; dfs(son[j]);
    }
    r[x]=tot;
}
void make_tree(int x,int l,int r){
    tree[x].l=l,tree[x].r=r,tree[x].tag=-1;
    if (l==r) return;
    int mid=(l+r)>>1;
    make_tree(x*2,l,mid); make_tree(x*2+1,mid+1,r);
}
void pushdown(int x){
    if (tree[x].l==tree[x].r||tree[x].tag==-1) return;
    tree[2*x].tag=tree[x].tag; tree[2*x].sum=tree[x].tag*(tree[2*x].r-tree[2*x].l+1);
    tree[2*x+1].tag=tree[x].tag; tree[2*x+1].sum=tree[x].tag*(tree[2*x+1].r-tree[2*x+1].l+1);
    tree[x].tag=-1;
}
void update(int x,int l,int r,int num){
    pushdown(x);
    if (tree[x].l==l&&tree[x].r==r) {
        tree[x].sum=num*(tree[x].r-tree[x].l+1);
        tree[x].tag=num; return;
    }
    int mid=(tree[x].l+tree[x].r)>>1;
    if (r<=mid) update(x*2,l,r,num);
    else if (l>mid) update(x*2+1,l,r,num);
    else {update(x*2,l,mid,num); update(x*2+1,mid+1,r,num);}
    tree[x].sum=tree[x*2].sum+tree[x*2+1].sum;
}
int query(int x,int l,int r){
    pushdown(x);
    if (tree[x].l==l&&tree[x].r==r) return tree[x].sum;
    int mid=(tree[x].l+tree[x].r)>>1;
    if (r<=mid) return query(x*2,l,r);
    else if (l>mid) return query(x*2+1,l,r);
    else return (query(x*2,l,mid)+query(x*2+1,mid+1,r));
}
int main(){
    scanf("%d",&n);
    for (int i=1;iint x,y; scanf("%d %d",&x,&y);
        add(x,y); add(y,x);
    }
    tot=0; dfs(1); make_tree(1,1,n);
    scanf("%d",&Q);
    while (Q--){
        int c,x; scanf("%d %d",&c,&x);
        if (c==1) {
            if (query(1,p[x],r[x])!=r[x]-p[x]+1&&fa[x]) update(1,p[fa[x]],p[fa[x]],0);
            update(1,p[x],r[x],1);
        } else if (c==2) update(1,p[x],p[x],0);
        else printf("%d\n",(query(1,p[x],r[x])==r[x]-p[x]+1));
    }
    return 0;
}