Assign the task（dfs序+线段树）

HDOJ 3974

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

即两种操作：

• C x 表示对员工x的当前任务的查询
• Tx y 表示公司将任务y分配给员工x

分析： 因为这道题学习了dfs序，参考这篇博客

DFS序就是将树形结构转化为线性结构，用dfs遍历一遍这棵树，进入到x节点有一个in时间戳，递归退出时有一个out时间戳。
x节点的两个时间戳之间遍历到的点，就是根为x的子树的所有节点，他们的dfs进入时间戳是递增的。
同时两个时间戳构成了一个区间，x节点在这段区间的最左端，这个区间就是一棵根节点为x的子树，对于区间的操作就是其他维护方式的应用了。

in[x]表示映射的DFS预处理出的线性结构，也就是说x是原始节点，in[x]是x节点的新位置。
num[t]表示第t个节点的编号，num[in[x]]表示的还是x。
num是新序列，in表示是新序列的下标，in[x]~out[x]是x为根结点的子树，划分为一个区间。

ll cnt=0;

void dfs(ll x) {
	in[x]=++cnt;
	//num[cnt]=x;  //生成新的线性结构
	for (int i=0; i<G[x].size(); i++) {
		ll t=G[x][i];
		dfs(t);
	}
	out[x]=cnt;
}

回到这道题，先把每个老板建一个vector，员工是其元素
从领导开始建立dfs序，就能转换成线性结构，找到区间用线段树维护

#include <iostream>
#include <cmath>
#include <iomanip>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
#include <stack>
#include <set>
#include <cstdio>
#include <algorithm>
#define INF 0x3f3f3f3f
#define PI acos(-1)
using namespace std;
typedef long long ll;
typedef pair<ll, ll> P;
const int mod=1e9+7;
const int maxn=5e4+10;

struct Node {
	ll l,r,lz;
} tree[maxn<<2];

ll n,m;
ll in[maxn],out[maxn];
ll vis[maxn];	//记录员工出现，没有出现过就是领导
vector <ll> G[maxn];  //老板管的员工
ll cnt;

void dfs(ll x) {
	in[x]=++cnt;
	for (int i=0; i<G[x].size(); i++) {
		ll t=G[x][i];
		dfs(t);
	}
	out[x]=cnt;
}

void push_down(ll p) {
	if (tree[p].lz!=-1) {
		tree[p<<1].lz=tree[p].lz;
		tree[(p<<1)|1].lz=tree[p].lz;
		tree[p].lz=-1;
	}
}

void build(ll p, ll l, ll r) {
	tree[p].l=l;
	tree[p].r=r;
	tree[p].lz=-1;
	if (l==r) {
		return;
	}
	ll m=(l+r)>>1;
	build(p<<1,l,m),build((p<<1)|1,m+1,r);
}

void update(ll p, ll x, ll y, ll z) {
	if (x<=tree[p].l && y>=tree[p].r) {
		tree[p].lz=z;
		return;
	}
	push_down(p);
	ll m=(tree[p].l+tree[p].r)>>1;
	if (x<=m) update(p<<1,x,y,z);
	if (y>m) update((p<<1)|1,x,y,z);
}

ll query(ll p, ll x) {
	if (tree[p].l == tree[p].r) {
		return tree[p].lz;
	}
	push_down(p);
	ll m=(tree[p].l+tree[p].r)>>1;
	if (x<=m) return query(p<<1,x);
	else  return query((p<<1)|1,x);
}

int main() {

	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	int T;
	cin>>T;
	int ca=1;
	while (T--) {		
		cin>>n;
		for (int i=0; i<=n; i++) {
			vis[i]=0;
			G[i].clear();
		}
		build(1,1,n);				
		for (int i=1; i<n; i++) {
			ll u,v;
			cin>>u>>v;
			G[v].push_back(u);
			vis[u]=1;
		}
		cnt=0;
		for (int i=1; i<=n; i++) {
			if (!vis[i]) {
				dfs(i);
				break;
			}
		}
		cin>>m;
		cout<<"Case #"<<ca++<<":"<<endl;
		char tag;
		ll x,y;
		while (m--) {
			cin>>tag;
			if (tag=='C') {
				cin>>x;
				cout<<query(1,in[x])<<endl;
			} else {
				cin>>x>>y;
				update(1,in[x],out[x],y);
			}
		}
	}
	return 0;
}

/********
1 
5 
4 3 
3 2 
1 3 
5 2 
5 
C 3 
T 2 1
C 3 
T 3 2 
C 3
**********/