L2-025 分而治之(并查集)

Posted 2021-02-04 taozi1115402474

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

输入格式：

输入在第一行给出两个正整数 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

题意

如上

题解

直接暴力并查集，模板

代码

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 
 4 const int maxn=10005;
 5 int u[maxn],v[maxn],f[maxn];
 6 bool d[maxn];
 7 int find(int x)
 8 {
 9     return f[x]==x?x:f[x]=find(f[x]);
10 }
11 int main()
12 {
13     int n,m;
14     scanf("%d%d",&n,&m);
15     for(int i=0;i<m;i++)
16         scanf("%d%d",&u[i],&v[i]);
17     int k;
18     scanf("%d",&k);
19     while(k--)
20     {
21         int np;
22         scanf("%d",&np);
23         for(int i=1;i<=n;i++)
24             d[i]=false,f[i]=i;
25         for(int i=0,x;i<np;i++)
26             scanf("%d",&x),d[x]=true;
27         for(int i=0;i<m;i++)
28             if(!d[u[i]]&&!d[v[i]])
29                 f[find(u[i])]=find(v[i]);
30         int ans=0;
31         for(int i=1;i<=n;i++)
32             if(f[i]==i)
33                 ans++;
34         printf("%s
",ans==n?"YES":"NO");
35     }
36     return 0;
37 }