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

有一家公司有N个员工(从1到N)，公司里每个员工都有一个直接的老板(除了整个公司的领导)。如果你是某人的直接老板，那个人就是你的下属，他的所有下属也都是你的下属。如果你是没有人的老板，那么你就没有下属，没有直接老板的员工就是整个公司的领导，也就是说N个员工构成了一棵树。公司通常把一些任务分配给一些员工来完成，当一项任务分配给某个人时，他/她会把它分配给他/她的所有下属，换句话说，这个人和他/她的所有下属在同一时间接受了一项任务。此外，每当员工收到一个任务，他/她将停止当前任务(如果他/她有)，并开始新的任务。在公司将某些任务分配给某个员工后，编写一个程序来帮助找出某个员工当前的任务。

Input第一行包含单个正整数T(T<=10)，表示测试用例的数量。对于每个测试用例：第一行包含一个整数N(N≤50，000)，它是雇员的数目。下面的N-1行分别包含两个整数u和v，这意味着雇员v是雇员u的直接老板(1<=u，v<=N)。下一行包含一个整数M(M≤50，000)。下面的M行分别包含一条消息，“Cx”表示对员工x的当前任务的查询，“Tx y”表示公司将任务y分配给员工x。(1<=x<=N，0<=y<=10^9)Output对于每个测试用例，在第一行打印测试用例编号(以1开头)，然后为每个查询输出相应的答案。Sample Input

1
5
4 3
3 2
1 3
5 2
5
C 3
T 2 1
C 3
T 3 2
C 3

Sample Output

Case #1:
-1
1
2

分析：用dfs遍历多叉树，按遍历顺序给每个人编号并保存子树的区间，然后直接套线段树的模板，进行区间的更新（子树的区间是连续的）；

代码：

 #include <bits/stdc++.h>
 using namespace std;
 const int maxn =  * 1e4 + ;
 struct bo
 {
     int l, r;
     int task;
 }t[maxn << ];

 int n;
 int clockk;
 bool vis[maxn];
 vector<int> bos[maxn];
 int l[maxn], r[maxn];

 void pushdown(int tar)
 {
     if (t[tar].task != -)
     {
         t[tar << ].task = t[tar <<  | ].task = t[tar].task;
         t[tar].task = -;
     }
 }

 void build(int l, int r, int tar)
 {
     t[tar].l = l, t[tar].r = r, t[tar].task = -;
     if (l == r) return;
     int mid = (l + r) >> ;
     build(l, mid, tar << );
     build(mid + , r, tar <<  | );
 }
 void dfs(int x)
 {
     l[x] = ++clockk;
     for (int i = ; i < bos[x].size(); i++)
         dfs(bos[x][i]);
     r[x] = clockk;
 }

 void update(int l, int r, int work, int tar)
 {
     if (t[tar].l == l && t[tar].r == r)
     {
         t[tar].task = work;
         return;
     }
     pushdown(tar);
     int mid = (t[tar].l + t[tar].r) >> ;
     if (r <= mid) update(l, r, work, tar << );
     else if (l > mid) update(l, r, work, tar <<  | );
     else update(l, mid, work, tar << ), update(mid + , r, work, tar <<  | );
 }

 int query(int x, int tar)
 {
     if (t[tar].l == t[tar].r) return t[tar].task;
     pushdown(tar);
     int mid = (t[tar].l + t[tar].r) >> ;
     if (x <= mid) return query(x, tar << );
     else return query(x, tar <<  | );
 }

 int main()
 {
     int T; cin >> T;
     int boss, emp;
     char ope[];
     int cases = ;

     while (T--)
     {
         cin >> n;
         clockk = ;
         memset(vis, , sizeof(vis));
         for (int i = ; i <= n; i++)
             bos[i].clear();
         build(, n, );
         for (int i = ; i < n; i++)
         {
             scanf("%d%d", &emp, &boss);
             vis[emp] = true;
             bos[boss].push_back(emp);
         }
         for (int i = ; i <= n; i++)
             if (!vis[i])
                 dfs(i);
         int m, x, y;

         cin >> m;
         printf("Case #%d:\n", ++cases);
         while (m--)
         {
             cin >> ope;
             if (ope[] == 'C')
             {
                 scanf("%d", &x);
                 cout << query(l[x], ) << endl;
             }
             else
             {
                 scanf("%d%d", &x, &y);
                 update(l[x], r[x], y, );
             }
         }
     }
 }