Water Tree

Codeforces Round #200 (Div. 1) D:http://codeforces.com/problemset/problem/343/D

题意：给你一颗有根树，树的每个节点有一个装水的东西，然后付亲的水可以流向儿子。一开始树上每个节点都是空的，然后会有两种操作。1 v，表示把v节点为子树的每个点都灌满水，2 v，表示v及v的祖先都清空。3 v是询问 v节点是否为空，为空输出0，否则输出1.

题解：首先，树上操作可以通过DFS，然后转化成数组操作。3操作来说，只要v的子树中有一个0，那么v一定是0，所以把单点查询，转化成了区间查询。2操作来说，我们可以只进行单点更新，即只对v这个节点操作。1操作的时候，就是区间更新，但是这里首先要查询v中是否有0，因为2操作只到了点。加入说，之前 2 v，那么v及其父亲应该是0的，这时候如果1，v的话，那么必须要把v的父亲变为0.这里有两种实现。先说一种好写。set运用。我们一开始把树上的每个节点都放进set，表示每个几点都是0.对于操作1.来说，首先我们要查询是否有0，Q.lower_bound(ll[temp]),如果找到并且这个数的位置不超过rr[temp],那么v的父亲要变成0,即可v的父亲查到set中，同时把那些变成1的元素从set中删除，Q.erase(Q.lower_bound(ll[temp]),Q.upper_bound(rr[temp]));3操作类似。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<set>
 using namespace std;
 const int N=5e5+;
 int head[N],now;
 int fa[N],ll[N],rr[N],cnt;
 set<int>Q;
 void init(){
      memset(head,-,sizeof(head));
      memset(rr,,sizeof(rr));
      memset(ll,,sizeof(ll));
      memset(fa,,sizeof(fa));
      now=;
      cnt=;
 }
 struct Node{
   int v;
   int next;
 }edge[N*];
 void add(int u,int v){
    edge[cnt].v=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
 }
 void DFS(int u,int f){
      ll[u]=rr[u]=++now;
      Q.insert(now);
      fa[u]=f;
      for(int i=head[u];i!=-;i=edge[i].next){
         int v=edge[i].v;
         if(v!=f)
             DFS(v,u);
      }
      rr[u]=now;
 }
 int n,m,t1,t2;
 int main(){
     while(~scanf("%d",&n)){
             init();
         for(int i=;i<n;i++){
             scanf("%d%d",&t1,&t2);
             add(t1,t2);add(t2,t1);
         }
         DFS(,);
         scanf("%d",&m);
         for(int i=;i<=m;i++){
             scanf("%d%d",&t1,&t2);
             if(t1==){
                   bool flag=false;
                 set<int>::iterator it=Q.lower_bound(ll[t2]);
                 if(it==Q.end()||(*it)>rr[t2])
                    flag=true;
             Q.erase(Q.lower_bound(ll[t2]),Q.upper_bound(rr[t2]));
                 if(!flag&&t2!=)
                     Q.insert(ll[fa[t2]]);
             }
             else if(t1==){
                 Q.insert(ll[t2]);
             }
             else {
                 set<int>::iterator it=Q.lower_bound(ll[t2]);
                 if(it==Q.end()||(*it)>rr[t2])
                    printf("1\n");
                    else
                     printf("0\n");
             }
         }
     }
 }

第二种实现：线段树版本。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<set>
 using namespace std;
 const int N=5e5+;
 int head[N],now;
 int fa[N],ll[N],rr[N],cnt;
 int n,m,t1,t2;
 int lazy[N*],seg[N*];
 void init(){
      memset(head,-,sizeof(head));
      memset(rr,,sizeof(rr));
      memset(ll,,sizeof(ll));
      memset(fa,,sizeof(fa));
      memset(lazy,,sizeof(lazy));
      memset(seg,,sizeof(seg));
      now=;
      cnt=;
 }
 struct Node{
   int v;
   int next;
 }edge[N*];
 void add(int u,int v){
    edge[cnt].v=v;
    edge[cnt].next=head[u];
    head[u]=cnt++;
 }
 void DFS(int u,int f){
      ll[u]=rr[u]=++now;
      fa[u]=f;
      for(int i=head[u];i!=-;i=edge[i].next){
         int v=edge[i].v;
         if(v!=f)
             DFS(v,u);
      }
      rr[u]=now;
 }
 void push_up(int rt){
   seg[rt]=min(seg[rt<<],seg[rt<<|]);
 }
 void pushdown(int rt){
   if(lazy[rt]>){
     lazy[rt<<]=lazy[rt<<|]=lazy[rt];
     seg[rt<<]=seg[rt<<|]=seg[rt];
     lazy[rt]=-;
   }
 }
 void update(int l,int r,int rt,int x,int y){
     if(l==x&&y==r){
         lazy[rt]=seg[rt]=;
         return;
     }
     pushdown(rt);
    int mid=(l+r)/;
    if(mid>=y)update(l,mid,rt<<,x,y);
    else if(mid<x)update(mid+,r,rt<<|,x,y);
    else {
       update(l,mid,rt<<,x,mid);
       update(mid+,r,rt<<|,mid+,y);
    }
    push_up(rt);
 }
 void update1(int l,int r,int rt,int x){
     if(l==r){
         lazy[rt]=seg[rt]=;
         return;
     }
    pushdown(rt);
    int mid=(l+r)/;
    if(mid>=x)update1(l,mid,rt<<,x);
    else update1(mid+,r,rt<<|,x);
    push_up(rt);
 }
 int query(int l,int r,int rt,int x,int y){
     if(l==x&&r==y){
         return seg[rt];
     }
     pushdown(rt);
     int mid=(l+r)/;
     if(mid>=y)return query(l,mid,rt<<,x,y);
     else if(mid<x) return query(mid+,r,rt<<|,x,y);
     else {
         return min(query(l,mid,rt<<,x,mid),query(mid+,r,rt<<|,mid+,y));
     }
 }

 int main(){
     while(~scanf("%d",&n)){
             init();
         for(int i=;i<n;i++){
             scanf("%d%d",&t1,&t2);
             add(t1,t2);add(t2,t1);
         }
         DFS(,);
         scanf("%d",&m);
         for(int i=;i<=m;i++){
             scanf("%d%d",&t1,&t2);
             if(t1==){
                 int temp=query(,now,,ll[t2],rr[t2]);
                 update(,now,,ll[t2],rr[t2]);
                 if(temp==&&t2!=)
                     update1(,now,,ll[fa[t2]]);
             }
             else if(t1==){
                  update1(,now,,ll[t2]);
             }
             else {
                 printf("%d\n",query(,now,,ll[t2],rr[t2]));
             }
         }
     }
 }