3-5 分而治之 (15 分)（并查集秒解）

3-5 分而治之 (15 分)

分而治之，各个击破是兵家常用的策略之一。在战争中，我们希望首先攻下敌方的部分城市，使其剩余的城市变成孤立无援，然后再分头各个击破。为此参谋部提供了若干打击方案。本题就请你编写程序，判断每个方案的可行性。

输入格式：

输入在第一行给出两个正整数 N 和 M（均不超过10 000），分别为敌方城市个数（于是默认城市从 1 到 N 编号）和连接两城市的通路条数。随后 M 行，每行给出一条通路所连接的两个城市的编号，其间以一个空格分隔。在城市信息之后给出参谋部的系列方案，即一个正整数 K （≤ 100）和随后的 K 行方案，每行按以下格式给出：

Np v[1] v[2] ... v[Np]

其中 Np 是该方案中计划攻下的城市数量，后面的系列 v[i] 是计划攻下的城市编号。

输出格式：

对每一套方案，如果可行就输出YES，否则输出NO。

输入样例：

10 11
8 7
6 8
4 5
8 4
8 1
1 2
1 4
9 8
9 1
1 10
2 4
5
4 10 3 8 4
6 6 1 7 5 4 9
3 1 8 4
2 2 8
7 9 8 7 6 5 4 2

输出样例：

NO
YES
YES
NO
NO

 

代码：

#include<iostream>
using namespace std;
int pre[10005];
int load[10005][2] = { 0 };


void init(int n)
{
    for (int i = 1; i <= n; i++)
    {
        pre[i] = i;
    }
}
int find(int a)
{
    if (pre[a] == a) return a;
    else return pre[a] = find(pre[a]);
}
void unit(int a, int b)
{
    int fa = find(a);
    int fb = find(b);
    if (a != b)
    {
        pre[fa] = fb;
    }
}
int main()
{
    int n, m, temp = 0;
    cin >> n >> m;
    for (int i = 0; i < m; i++)
    {
        cin >> load[i][0] >> load[i][1];
    }
   
    int k;
    cin >> k;
    for (int i = 0; i < k; i++)
    {
        int tag[10005] = { 0 };
        bool flag[10005] = { false };
        init(n);
        int p, city;
        cin >> p;
        for (int j = 0; j < p; j++)
        {
            cin >> city;
            flag[city] = true;
        }
        for (int i = 0; i < m; i++)
        {
            if (flag[load[i][0]] == false && flag[load[i][1]] == false)
            {
                unit(load[i][0], load[i][1]);
            }
        }
        for (int i = 1; i <= n; i++)
        {
            if (tag[find(i)] == 1)
            {
                temp = 1;
                break;
            }
            else tag[find(i)] = 1;
        }
        if (temp == 1) cout << "NO" << endl;
        else cout << "YES" << endl;
        temp = 0;
        //bool flag[10005] = { false };
        //int tag[10005] = { 0 };

    }
}