HDU-3974 Assign the task (dfs序+线段树)

链接 hdu-3974
题意：
给一颗树，每次给一个节点的所有子节点分配一个值，值会被覆盖，询问每次某个节点的值，若没有分配，输出-1；
思路：
dfs建树裸题了，建树后权值相当于 lazy标记，区间更新，单点查询就好了。
这题我wa了不下60遍，也不知道为什么跟个sb一样。

1. 首先 多组输入，多组输入！！！！！。

 memset(head,0,sizeof(head));
        cnt=0;num=0;
        for(int i=0;i<=n;i++) degree[i]=0;

2. 然后查询记得用 dfs序啊 ！！！我居然直接查了x
3. 终究是码力不够，浪费了几个小时的时间

#include <cstdio>
#include <iostream>
#include <cstring>
#include <vector>
#include <cmath>
#include <algorithm>
#include <map>
#include <queue>
#include <cmath>
#include <ctime>
using namespace std;
typedef long long ll;
const int maxn=1e6+7;
int n,m,head[maxn],num=0,root;
int ans,val[maxn],la[maxn<<2],degree[maxn];
struct node{
    int next,to;
};
struct node e[maxn<<1];
void add(int u,int v){
     e[++num].next=head[u];
     e[num].to=v;
     head[u]=num;
}
int cnt,l[maxn],r[maxn];
void dfs(int u, int pre){
    l[u]=++cnt;
    for(int i = head[u]; i; i = e[i].next){
        int to = e[i].to;
        if(to == pre) continue;
        dfs(to, u);
    }
    r[u]=cnt;
}
void pushdown(int rt){
     if(la[rt]!=0){
        la[rt<<1]=la[rt];
        la[rt<<1|1]=la[rt];
        la[rt]=0;
     }
}
void update1(int le,int ri,int v,int l,int r,int rt){
     if(le<=l&&ri>=r){
        la[rt]=v;
        return ;
     }
     pushdown(rt);
     ll m=(l+r)>>1;

     if(le<=m) update1(le,ri,v,l,m,rt<<1);

     if(ri>m) update1(le,ri,v,m+1,r,rt<<1|1);

}
int query(int id,int l,int r,int rt){
    if(la[rt]!=0) return la[rt];
    pushdown(rt);
    ll m=(l+r)>>1;
    if(id<=m)  return query(id,l,m,rt<<1);
    if(id>m)   return query(id,m+1,r,rt<<1|1);

}
int main(){
   int t;
   cin>>t;
   int ca=1;
   while(t--){
        int n,u,v,m,x,y;
        cin>>n;
        memset(head,0,sizeof(head));
        cnt=0;num=0;
        for(int i=0;i<=n;i++) degree[i]=0;
        for(int i=0;i<n-1;i++){
            scanf("%d%d",&u,&v);
            add(u,v);
            add(v,u);
            degree[u]++;
        }
        for(int i=1;i<=n;i++){
            if(!degree[i]){
                root=i;
                break;
            }
        }
        cout<<"Case #"<<ca++<<':'<<endl;
        dfs(root,-1);
        update1(l[root],r[root],-1,1,n,1);
        cin>>m;
        while(m--){
            char st[3];
            scanf("%s",st);
            if(st[0]=='C'){
                scanf("%d",&x);
                cout<<query(l[x],1,n,1)<<endl;
            }
            else{
                scanf("%d%d",&x,&y);
                update1(l[x],r[x],y,1,n,1);
            }
        }
   }
}