hdu 3974 Assign the task（线段树+将树映射到区间）

Assign the task

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=3974

解题思路：

题目大意：

一家公司有n个人，编号1～n，每个人都有一个直属上司。 

给你两种操作： 

1：给x分配任务y，此时，x会将y也分配给他的下属，下属再分配给下属……也就是x下面的人都会放下原先的任务（如果有）而去

做任务y。

2：询问x当前的任务，若没有则输出-1。

算法思想：

将树映射到区间的线段树。。。

主要是将原有的关系树根据BOSS关系从新编号，

以便把每个BOSS所带领的员工全部压入一个连续区间内，

然后记录每个BOSS的起始编号和他的最后一名的员工的编号，

然后用线段树成端更新，单点查找即可。

AC代码：

#include <bits/stdc++.h>using namespace std;const int maxn = 50005;int fa[maxn],cnt;int sta[maxn],tail[maxn],num[maxn],curTask[maxn];vector<int> g[maxn];struct node{    int l,r,setv;}tree[maxn<<2];void dfs(int u,int fa){    num[u] = ++cnt;    curTask[u] = num[u];    sta[u] = cnt;//BOSS的起始编号    int len = g[u].size();    for(int i = 0; i < len; ++i){        if(g[u][i] == fa)            continue;        dfs(g[u][i],u);    }    tail[u] = cnt;//该BOSS的最后一名员工的编号}void pushdown(int id){    if(tree[id].setv != -1){        tree[id<<1].setv = tree[id<<1|1].setv = tree[id].setv;        tree[id].setv = -1;    }}void build(int id,int l,int r){    tree[id].l = l;    tree[id].r = r;    tree[id].setv = -1;    if(l == r){        return;    }    int mid = (l+r)>>1;    build(id<<1,l,mid);    build(id<<1|1,mid+1,r);}void update(int id,int l,int r,int val){    if(tree[id].l >= l && tree[id].r <= r){        tree[id].setv = val;        return;    }    int mid = (tree[id].l+tree[id].r)>>1;    pushdown(id);    if(l <= mid)        update(id<<1,l,r,val);    if(r > mid)        update(id<<1|1,l,r,val);}int query(int id,int pos){    if(tree[id].l == tree[id].r)        return tree[id].setv;    int mid = (tree[id].l+tree[id].r)>>1;    pushdown(id);    if(pos <= mid)        return query(id<<1,pos);    else        return query(id<<1|1,pos);}int main(){    int T,t = 1;    scanf("%d",&T);    while(T--){        printf("Case #%d:\n",t++);        int n,q;        scanf("%d",&n);        for(int i = 0; i <= n; ++i)            g[i].clear();        memset(fa,0,sizeof(fa));        for(int i=1; i<n; ++i){            int u,v;            scanf("%d%d",&u,&v);            fa[u] = v;            g[u].push_back(v);            g[v].push_back(u);        }        cnt = 0;        for(int i=1; i<=n; ++i){            if(fa[i] == 0){                dfs(i,0);                break;            }        }        build(1,1,n);        scanf("%d",&q);        while(q--){            char op[5];            scanf("%s",op);            int x,y;            scanf("%d",&x);            if(op[0]=='C'){                printf("%d\n",query(1,curTask[x]));            }            else{                scanf("%d",&y);                update(1,sta[x],tail[x],y);            }        }    }    return 0;}