hdu5923(并查集维护联通块)

前一篇博文说这题t了，今天看了一下别人的写法发现有个地方复杂度确实常数有点大。

对magic tree上的每个点维护它到根的点的集合（并且是原图的生成树，这样点集的大小是O(n)的),这样的话之后询问的时候每个点考虑的点数就少了很多。
另外在dfs预处理的时候要充分运用父亲的状态，重新并一遍就会tle。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <vector>

using namespace std;

const int maxn = 505;
const int maxm = 10005;

int f[maxm][maxn];

vector<int> G[maxm];

int n,m,q;
int from[maxm],to[maxm];

int Find(int x,int ty){
    return x == f[ty][x] ? x : f[ty][x] = Find(f[ty][x],ty);
}

void dfs(int u,int fa){
    for(int i = 1;i <= n;i++){
        f[u][i] = f[fa][i];
    }
    int fx = Find(from[u],u);
    int fy = Find(to[u],u);
    if(fx != fy){
        f[u][fx] = fy;
    }
    for(int i = 0;i < G[u].size();i++){
        int v = G[u][i];
        dfs(v,u);
    }
}

int main(){
    int T;
    int cas = 0;
    cin >> T;
    while(T--){
        cin >> n >> m;
        for(int i = 1;i <= m;i++) G[i].clear();
        for(int i = 2;i <= m;i++){
            int par;
            scanf("%d",&par);
            G[par].push_back(i);
        }
        for(int i = 1;i <= n;i++) f[1][i] = i;
        for(int i = 1;i <= m;i++){
            scanf("%d%d",from + i,to + i);
        }
        dfs(1,1);
        scanf("%d",&q);
        printf("Case #%d:\n",++cas);
        for(int i = 1;i <= q;i++){
            int ki;
            scanf("%d",&ki);
            for(int j = 1;j <= n;j++){
                f[0][j] = j;
            }
            int num = n;
            for(int j = 1;j <= ki;j++){
                int a;scanf("%d",&a);
                for(int k = 1;k <= n;k++){
                    int ty = Find(k,a);
                    if(ty != k){
                        int x = Find(ty,0);
                        int y = Find(k,0);
                        if(x != y){
                            num--;
                            f[0][x] = y;
                        }
                    }
                }
            }
            printf("%d\n",num);
        }
    }
    return 0;
}