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;}