HDU3974 Assign the task（多叉树转换为线段+线段树区间染色）

题目大意：有n个人，给你他们的关系（老板和员工），没有直属上司的人就是整个公司的领导者，这意味着n个人形成一棵树(多叉树)。当一个人被分配工作时他会让他的下属也做同样的工作（并且立即停止手头正在做的工作），题目会询问你其中某个人正在做的工作。

解题思路：其实从“一个人分配他的下属做一样的工作”这里就可以看出来了，这相当于让一块区间的人都做一样的事，就是线段树区间染色问题。但不能使用线段树，要先将多叉树铺展开，将节点映射到线段上。把每个人的管理区段找出来（把属于同一个人管的放一起，上司放在前面），这样对某个员工更新也就是对他和他下属的更新。具体实现就是先将多叉树保存下来，用dfs遍历多叉树给每个人打上时间戳，分配序号就行了。

#include<iostream>
#include<cstring>
#include<vector>
#define LC(a) ((a<<1))
#define RC(a) ((a<<1)+1)
#define MID(a,b) ((a+b)>>1)
using namespace std;
const int N=5e4+5;
typedef long long ll;

vector<ll>v[N];
ll Start[N],End[N];//每个员工所有下属的开始和结束节点，包含本身
ll ans,cnt;//cnt用于记录节点的编号
bool used[N];

void dfs(ll rt){
    Start[rt]=++cnt;
    for(int i=0;i<v[rt].size();i++){
        dfs(v[rt][i]);
    }
    End[rt]=cnt;
}

struct node{
    ll l,r;
    ll task;//task=-2表示下属工作不同 
}tree[N*4];

void pushup(ll p){
    tree[p].task=(tree[LC(p)].task==tree[RC(p)].task?tree[LC(p)].task:-2);
}

void pushdown(ll p){
    tree[LC(p)].task=tree[RC(p)].task=tree[p].task;
}

void build(ll p,ll l,ll r){
    tree[p].l=l;
    tree[p].r=r;
    tree[p].task=-1;
    if(l==r){
        return;
    }
    build(LC(p),l,MID(l,r));
    build(RC(p),MID(l,r)+1,r);
}

void update(ll p,ll l,ll r,ll task){
    if(r<tree[p].l||l>tree[p].r)
        return;
    if(l<=tree[p].l&&r>=tree[p].r){
        tree[p].task=task;
        return;
    }
    if(tree[p].task!=-2)
        pushdown(p);
    update(LC(p),l,r,task);
    update(RC(p),l,r,task);
    pushup(p);
}

void query(ll p,ll t){
    if(tree[p].task!=-2){
        ans=tree[p].task;
        return;
    }
    ll mid=MID(tree[p].l,tree[p].r);
    if(t<=mid)
        query(LC(p),t);
    else
        query(RC(p),t);
}

int main(){
    ios::sync_with_stdio(false);
    ll t;
    ll cas=0;
    cin>>t;
    while(t--){
        cas++;
        //初始化 
        cnt=0; 
        memset(used,false,sizeof(used));
        for(int i=1;i<=N;i++){
            v[i].clear();
        }
        
        ll n;
        cin>>n;
        for(int i=1;i<=n-1;i++){
            ll    rt,chd;
            cin>>chd>>rt;
            used[chd]=true;
            v[rt].push_back(chd);
        }
        //将多叉树转化为线段 
        for(int i=1;i<=n;i++){
            //找到根结点 
            if(!used[i]){
                dfs(i);
                break;
            }
        }
        //建树 
        build(1,1,n); 
        ll m;
        cout<<"Case #"<<cas<<":"<<endl;
        cin>>m;    
        for(int i=1;i<=m;i++){
            char op;
            cin>>op;
            if(op=='C'){
                ll x,t;
                cin>>x;
                t=Start[x];
                query(1,t);
                cout<<ans<<endl;
            }
            else{
                ll x,l,r,task;
                cin>>x>>task;
                l=Start[x];
                r=End[x];
                update(1,l,r,task);
            }
        }            
    }
}