JZOJ3883 [NOIP2014模拟] 线段树(segment)

Description
给定一个n个点n-1条边的无向连通图（一棵树），并对图中的边进行m次染色操作。
每次染色操作给定2个点u、v和一种颜色c，并将图中u，v之间的最短路上的边都染成这种颜色。
询问的是最终图中每条边的颜色。（若未被染色则视为颜色0即无色）

Input
第1行：2个整数n，m
第2到第n行：第i行表示i号点的父节点fi
接下来m行：每行3个整数ui，vi，ci分别表示给定的节点u、v和颜色c。

Output
共n-1行：第i行输出第i+1个点与其父节点之间边的颜色。

Data Constraint
对于10%的数据，n,m<=100
对于30%的数据，n,m<=5000
对于60%的数据，n,m<=100000
对于100%的数据，n,m<=500000,0<ci<=m

吐槽：
话说这玩意跟线段树半毛钱关系都没有啊，这不误人子弟吗！

Solution
我们注意到对于每一条边对它有影响的只有最晚的一次染色，所以我们可以将染色倒序处理，
每次染色将树上 $u,v$ 两点间所有未染色的边染色，同时用并查集将这一段路径都缩成一个
团并记录每个团在树上最浅的点。由于每条边只会被染色一次并且之后就并入团中不会再被
访问到，可以用BFS，故而只有 $O（m+n）$次的并查集操作，即可通过此题。
时间复杂度：$O(m+n\ast \alpha (n))$ 空间复杂度：$O（n+m）$

Code

#include <cstdio>
using namespace std;

#define N 500007
int fa[N], fir[N], nxt[N], f[N], c[N], dep[N];
int qry[N][3], data[N];
bool bb;

void inc(int x, int y){nxt[y] = fir[x], fir[x] = y;}

void bfs()
{
	int head = 0, tail = 1, u;
	data[1] = dep[1] = 1;
	while (head < tail)
	{
		u = data[++ head];
		for (int i = fir[u]; i; i = nxt[i])
			dep[i] = dep[u] + 1, data[++ tail] = i;
	}
}

int find(int x)
{
	return (f[x] == x) ? (x) : (f[x] = find(f[x]));
}

void color(int u, int v, int C)
{
	u = find(u), v = find(v);
	while (u != v)
	{
		if (dep[u] < dep[v]) u ^= v, v ^= u, u ^= v;
		c[u] = C, u = f[u] = find(fa[u]);
	}
}

int main()
{
	int n, m;
	scanf("%d%d", &n, &m);
	for (int i = 2; i <= n; ++ i) scanf("%d", fa + i), inc(fa[i], i);
	for (int i = m; i; -- i) scanf("%d%d%d", &qry[i][0], &qry[i][1], &qry[i][2]);
	for (int i = 1; i <= n; ++ i) f[i] = i;
	bfs();
	for (int i = 1; i <= m; ++ i) color(qry[i][0], qry[i][1], qry[i][2]);
	for (int i = 2; i <= n; ++ i) printf("%d\n", c[i]);
	return 0;
}