Assign the task (并查集)

题目是这样的：公司里有N个职工（公司老板也是其中一名）（1--N编号），职工分等级，公司老板是终极大boss，管理所有员工，然后分层，每层都有若干boss，手底下有若干手下，直到最底层的无产阶级；除了大boss以外，每个boss都有上司，即boss的boss， 而boss的手下也属于boss的上司的手下；现在开始分配任务，当一个人分配到任务后，他所有的手下都停下手头工作，来处理这个工作；问x号员工手头的工作是什么；

并查集就好了，不过需要开一个存时间的数组，而且不能压缩路径；

上代码：

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
#define MAX 50000+5
using namespace std;
int uset[MAX];
int t[MAX];                    //存时间：t[2]=4表示2号手头的工作是4时间开始的（后面的工作覆盖前面的工作);
int ans[MAX];                  //存工作；ans[3]=1表示3号手头工作是1号工作；
int findd(int x){
    int r=x;
    int ed=x;                  //ed存x的boss中工作时间最晚的；
    int time=t[x];             
    while(uset[r]!=r){
        if(t[uset[r]]>=time){  //r的boss工作时间大于等于r的工作时间，r开始干boss的工作；
            time=t[uset[r]];
            ed=uset[r];
        }
        r=uset[r];
    }
    return ed;
}
void query(int x){          //打印x的工作；
    printf("%d\n",ans[findd(x)]);
}
void ass(int x, int y){   //x的工作是y；
   ans[x]=y;
}
int main(){
    int T;
    scanf("%d",&T);
    int cnt=0;
    while(T--){
        cnt++;
        memset(ans, -1, sizeof(ans));
        memset(t, 0, sizeof(t));
        int n;
        scanf("%d",&n);
        for(int i=1; i<=n; i++)
            uset[i]=i;
        for(int i=1; i<n; i++){
            int son, father;
            scanf("%d %d",&son,&father);
            uset[son]=father;
        }
        printf("Case #%d:\n",cnt);
        int m;
        scanf("%d",&m);
        for(int i=1; i<=m; i++){
            char ch[5];              //由于输入单个字符，后边还有空格，所以直接输入字符串，不用考虑字符输入的处理；
            int x, y;
            scanf("%s",ch);
            if(ch[0]=='C'){
                scanf("%d",&x);
                query(x);
            }
            else{
                scanf("%d%d",&x,&y);
                t[x]=i;
                ass(x,y);
            }
        }
    }
    return 0;
}