HDU 3974 Assign the task 并查集

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

题目大意：

一个公司有N个员工，对于每个员工，如果他们有下属，那么他们下属的下属也是他的下属。

公司会给员工安排任务，分配给一个员工后，他也会把这个任务分配给下属。被分配到任务的人立刻停止 当前在做的工作，接受新的任务。

对于给定的M个操作

C x 输出编号为x的任务

T x y 分配任务y给x

思路：

并查集的实现，分配我们只记录在上司结点里，只不过查询的时候要把它的所有上司全部找一遍。

#include<cstdio>
#include<iostream>
using namespace std;
const int MAXN=50001;
int fa[MAXN];
int val[MAXN];
int time[MAXN];
int main()
{
	int T;
	scanf("%d",&T);
	for(int ri=1;ri<=T;ri++)
	{
		int n;
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
				fa[i]=i;
				val[i]=-1;
				time[i]=-1;
		}

		for(int i=1;i<n;i++)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			fa[a]=b;
		}

		printf("Case #%d:\n",ri);
		int m,cnt=0;
		scanf("%d",&m);
		char cmd;
		while(m--)
		{
			cin>>cmd;
			if(cmd=='C')
			{
				int x;
				scanf("%d",&x);
				int cur=x,ans=val[cur],lastt=-1;
				while(cur != fa[cur])
				{
					if(time[cur] > lastt)//后加入的
					{
						lastt=time[cur];
						ans=val[cur];
					}
					cur=fa[cur];				
				}

				if(time[cur] > lastt)//后加入的
				{
					lastt=time[cur];
					ans=val[cur];
				}
				printf("%d\n",ans);
			}
			else
			{
				int x,y;
				scanf("%d%d",&x,&y);
				val[x]=y;
				time[x]=cnt++;
			}
		}

	}
	return 0;
}