HDU3974 - Assign the task - 并查集

题意

给出$T$组数据，每组数据首先给出一个$n$，

接下去有$n-1$行数据，每行给出$x$和$y$，代表$x$的老板是$y$，

再给出一个$m$，表示$m$次询问，每次询问如果是 $C$ 和$x$，表示输出$x$当前的任务；

如果询问的是 $T$ 和$x$、$y$，表示把$y$任务交给$x$。

思路

这题可以用线段树写，也可以直接暴力（我用的暴力写法一直TLE），我用的是并查集写。

AC代码

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
#include<cmath>
#include<list>
#include<stdlib.h>
#include<map>
#include<stack>
#include<stdio.h>
#include<queue>
using namespace std;
typedef long long ll;
#define sc(T) scanf("%d",&T)
#define scc(x,y) scanf("%d %d",&x,&y)
#define pr(T) printf("%d\n",T)
#define f(a,b,c) for (int a=b;a<c;a++)
#define ff(a,b,c) for (int a=b;a>c;a--)
#define inf 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
#define eps 1e-9
#define PI acos(-1)

int f[50020];
struct node
{
    int job,time;
} a[50020];

int main()
{
    int T,n,m,cas=1;
    sc(T);
    while(T--)
    {
        sc(n);
        int x,y,cnt=0;
        for(int i=1; i<=n; i++)
            f[i]=a[i].job=a[i].time=-1;
        for(int i=1; i<=n-1; i++)
        {
            scc(x,y);
            f[x]=y;
        }
        printf("Case #%d:\n",cas++);
        sc(m);
        f(i,0,m)
        {
            char s;
            cin>>s;
            int x,y;
            if(s=='C') //询问x的当前任务
            {
                cin>>x;
                int y=a[x].job,t=a[x].time;
                while(x!=-1)
                {
                    if(a[x].time>t)
                        y=a[x].job,t=a[x].time;
                    x=f[x];
                }
                cout<<y<<endl;
            }
            else
            {
                cin>>x>>y;
                a[x].job=y;
                a[x].time=++cnt;
            }
        }
    }
    return 0;
}