HDU - 5923 Prediction(2016CCPC东北地区大学生程序设计竞赛)

HDU - 5923 Prediction(2016CCPC东北地区大学生程序设计竞赛)

传送门HDU - 5923

题意

一开始先给你$n,m$，代表一个图的点数和magic树的点数。
然后再给你一棵magic树，这个树每个节点存的是图中的一条边，所有节点存的边的连接的点数最大不超过500。
每次询问会给你几个在magic树上的点，问你把这些点及其他们所有的祖先点所代表的边连上，那连出来的图，有多少个连通块。

题解

这道题是求联通块，那么第一时间就想到了并查集，然后就让我想到了我以前写过的一道莫队题CodeChef - GERALD07，当然内道莫队正解是lct。这两道题一样，都是用n个并查集维护。
不跑题了，还是说这道题。
图最多只500个点，所以全都直接暴力就行。
每个节点一个并查集，表示当前节点及其他所有的祖先节点代表的边连接成的图。首先先用个dfs把这个求出来。
然后就很简单了，每次询问中，把询问的点的并查集全都连接到一起，最后循环求一下连通图个数就ok了。
ps：比赛时我并查集写了按序合并，因为数组太大mle了一发，写的时候要注意一下这题不能按序合并。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cstring>
#include <string>
#include <vector>
#include <stack>
#include <cmath>
#include <ctime>
#include <map>
#include <set>
#include <iomanip>
//#include <unordered_map>

//#pragma comment(linker, "/STACK:102400000,102400000")
#define fir first
#define sec second
#define lson l, mid, rt << 1
#define rson mid + 1, r, rt << 1 | 1
#define clr(x) memset(x,0,sizeof(x))
#define cld(x) memset(x,-1,sizeof(x))
#define clx(x) memset(x,63,sizeof(x))
#define cln(x) memset(x,-64,sizeof(x))
#define rush() int T;scanf("%d",&T);while(T--)
#define pi 3.1415926
#define VM 100047
#define EM 400047
#define rd(x) scanf("%d",&x);
#define seed() srand((unsigned)time(NULL))
#define random(a, b) rand() % (b - a + 1) + a
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;

const int inf = 0x3f3f3f3f;
const ll llf = 0x3f3f3f3f3f3f3f3f;
const int maxn = (int) 1e4 + 7;
const double eps = 1e-10;
const ll mod1 = (int) 1e9 + 7;
const ll mod2 = 998244353;
const ll has = 99959;
const int dx[] = {0, 1, 0, -1};
const int dy[] = {1, 0, -1, 0};
int n, m;
vector<int> vec[maxn];

struct rec {
    int x, y;

    rec(int _x = 0, int _y = 0) {
        x = _x;
        y = _y;
    }
} a[maxn];


bool vis[507];

struct UFC {
    int par[507];// rank[507];

    inline void init(int n) {
        //clr(rank);
        clr(par);
        for (int i = 0; i <= n; i++) {
            par[i] = i;
        }
    }

    inline int find(int x) {
        if (par[x] == x) {
            return x;
        } else {
            return par[x] = find(par[x]);
        }
    }

    inline void unite(int x, int y) {
        x = find(x);
        y = find(y);
        if (x == y) return;

        par[x] = y;

    }

    inline bool same(int x, int y) {
        return find(x) == find(y);
    }

    inline void mov(UFC s) {
        for (int i = 1; i <= 500; ++i) {
            par[i] = s.par[i];
        }
    }

    inline void ins(UFC s) {
        for (int i = 1; i <= n; ++i) {
            unite(s.find(i), i);
        }
    }
} s[maxn], ss;

void dfs(int u) {
    for (int i = 0; i < vec[u].size(); ++i) {
        int v = vec[u][i];
        s[v].mov(s[u]);
        s[v].unite(a[v].x, a[v].y);
        dfs(v);
    }
}

int main() {
    std::ios::sync_with_stdio(false);
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    int NUM = 0;
    rush() {
        clr(vec);
        printf("Case #%d:\n", ++NUM);
        scanf("%d%d", &n, &m);
        for (int i = 1; i <= m; ++i) {
            s[i].init(n);
        }
        for (int i = 2; i <= m; ++i) {
            int u;
            scanf("%d", &u);
            vec[u].push_back(i);
        }
        for (int i = 1; i <= m; ++i) {
            int x, y;
            scanf("%d%d", &x, &y);
            a[i].x = x;
            a[i].y = y;
        }
        s[1].unite(a[1].x, a[1].y);
        dfs(1);
        int q, p;
        scanf("%d", &q);
        while (q--) {
            scanf("%d", &p);
            int x;
            scanf("%d", &x);
            ss.mov(s[x]);
            for (int i = 2; i <= p; ++i) {
                scanf("%d", &x);
                ss.ins(s[x]);
            }
            int ans = 0;
            clr(vis);
            for (int i = 1; i <= n; ++i) {
                if (!vis[ss.find(i)]) {
                    ans++;
                    vis[ss.find(i)] = true;
                }
            }
            printf("%d\n", ans);
        }
    }

    return 0;
}