Assign the task(线段树 + DFS序)

题目链接：Assign the task(HDU - 3974)

题目大意：

现有一棵树，有以下操作：
1 . 节点x及其所有子孙颜色都变更为k。
2 . 要求你回答节点x的颜色。

解题思路：

对于题目要求，我们需要维护的是一棵树，我们知道树是一种树形结构而非线性结构，因此操作起来比较复杂。那么我们是否可以将树形结构转换成线性结构那，答案是肯定的。DFS序就是一种很好的方法。

DFS序是将树形结构转化为线性结构的一种方式，用dfs从根节点遍历一遍这棵树，进入到x节点时有一个in时间戳，递归退出时有一个out 时间戳，x节点的两个时间戳之间遍历到的点，就是根为x的子树的所有节点，他们的dfs进入时间戳是递增的。同时两个时间戳构成了一个区间[ in[x] , out[x] ]，x节点在这段区间的最左端，这个区间就是一棵根节点为x的子树，区间的长度就为以x为根节点的子树大小(包括x节点)。

通过DFS序，操作1就成了区间修改，操作2就成了单点查询，因此我们可以借助线段树来做。

AC代码：

#include <bits/stdc++.h>
using namespace std;
const int N = 5e4 + 10;
int n, q, cnt;
int in[N], out[N], du[N];
vector<int> edge[N];
struct node
{
	int l, r;
	int colour;
}tr[N << 2];
void pushdown(int u)
{
	tr[u << 1].colour = tr[u << 1 | 1].colour = tr[u].colour;
	tr[u].colour = 0;
}
void build(int l, int r, int u = 1)
{
	tr[u] = { l,r,0 };
	if (l == r) return;
	int mid = l + r >> 1;
	build(l, mid, u << 1), build(mid + 1, r, u << 1 | 1);
}
void modify(int l, int r, int c, int u = 1)
{
	if (tr[u].l >= l && tr[u].r <= r) tr[u].colour = c;
	else
	{
		if (tr[u].colour != 0) pushdown(u);
		int mid = tr[u].l + tr[u].r >> 1;
		if (l <= mid) modify(l, r, c, u << 1);
		if (r > mid) modify(l, r, c, u << 1 | 1);
	}
}
int query(int x, int u = 1)
{
	if (tr[u].l == tr[u].r) return tr[u].colour;
	if (tr[u].colour != 0) pushdown(u);
	int mid = tr[u].l + tr[u].r >> 1;
	return query(x, u << 1 | (x > mid));
}
void dfs(int x)
{
	in[x] = ++cnt;
	for (int i = 0; i < edge[x].size(); i++) dfs(edge[x][i]);
	out[x] = cnt;
}
void init()
{
	cnt = 0;
	for (int i = 0; i <= n; i++)
	{
		in[i] = out[i] = du[i] = 0;
		edge[i].clear();
	}
}
int main()
{
	int t; cin >> t;
	for (int j = 1; j <= t; j++)
	{
		printf("Case #%d:\n", j);
		scanf("%d", &n); init();
		for (int i = 1; i < n; i++)
		{
			int u, v; scanf("%d %d", &u, &v);
			edge[v].push_back(u);
			du[u]++;
		}
		int root;
		for (int i = 1; i <= n; i++) if (!du[i]) root = i;
		dfs(root); build(1, n);
		scanf("%d", &q);
		while (q--)
		{
			char op[2]; int x, k;
			scanf("%s%d", op, &x);
			if (op[0] == 'T')
			{
				scanf("%d", &k);
				modify(in[x], out[x], ++k);
			}
			else printf("%d\n", query(in[x]) - 1);
		}
	}
	return 0;
}