HDU3974 Assign the task

题目链接：https://vjudge.net/problem/HDU-3974

知识点：　　线段树、DFS

题目大意：

　　公司中有(N)个职员，一个职员对应一个上司，这(N)个职员刚好组成一棵树。当给一个职员(x)分配任务(y)时（即 (T) (x) (y))，(x)及其下属及其下属的下属都会放下手头的工作去完成任务(y)。对于每次询问（即(C) (x)），程序输出职员(x)正在进行什么任务。

解题思路：

　　先将公司中职员之间的关系抽象成一棵树，每次分配任务其实就是更新一棵子树。当我们按(DFS)遍历的顺序为一棵子树编号时，我们会发现它们是连续的一串数字。于是我们可以先给整棵树用(DFS)序编号。每次更新任务其实就是更新一串连续数字对应的职员的当前任务，这就可以用线段树来处理了；而查询职员的当前任务其实就是线段树的单点查询。

AC代码：

#include <cstdio>
#include <vector>

using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1

const int maxn = 50005;
vector<int> son[maxn];
bool root[maxn];
int cnt;
int lef[maxn],rit[maxn];
int now[maxn<<2];
bool lazy[maxn<<2];

void dfs(int rt){
    cnt++;
    lef[rt]=cnt;
    for(int i=0;i<son[rt].size();i++)
        dfs(son[rt][i]);
    rit[rt]=cnt;
}
void pushdown(int rt){
    now[rt<<1]=now[rt<<1|1]=now[rt];
    lazy[rt<<1]=lazy[rt<<1|1]=true;
    lazy[rt]=false;
}
void build(int l,int r,int rt){
    now[rt]=-1;
    lazy[rt]=false;
    if(r==l) return;
    int m=(l+r)>>1;
    build(lson);
    build(rson);
}
void update(int L,int R,int c,int l,int r,int rt){
    if(l!=r&&lazy[rt])    pushdown(rt);
    if(L==l&&R==r){
        now[rt]=c;
        lazy[rt]=true;
        return;
    }
    int m=(l+r)>>1;
    if(m>=L)    update(L,min(m,R),c,lson);
    if(m<R)     update(max(L,m+1),R,c,rson);
}
int query(int pos,int l,int r,int rt){
    if(l!=r&&lazy[rt])    pushdown(rt);
    if(l==r)  return now[rt];
    int m=(l+r)>>1;
    if(m>=pos)  return query(pos,lson);
    else    return query(pos,rson);
}
int main()
{
    int T,N,M,x,y,u,v;
    scanf("%d",&T);
    for(int t=1;t<=T;t++){
        printf("Case #%d:
",t);
        scanf("%d",&N);
        for(int i=1;i<=N;i++){
            root[i]=true;
            son[i].clear();
        }
        for(int i=1;i<N;i++){
            scanf("%d%d",&u,&v);
            son[v].push_back(u);
            root[u]=false;
        }
        cnt=0;
        for(int i=1;i<=N;i++){
            if(root[i]){
                dfs(i);
                break;
            }
        }
        build(1,cnt+1,1);
        
        scanf("%d",&M);
        char ord[5];
        for(int i=0;i<M;i++){
            scanf("%s",ord);
            if(ord[0]=='C'){
                scanf("%d",&x);
                printf("%d
",query(lef[x],1,cnt+1,1));
            }
            else{
                scanf("%d%d",&x,&y);
                update(lef[x],rit[x],y,1,cnt+1,1);
            }
        }
    }
    return 0;
}