分而治之pta--并查集和图的连通块(团体程序设计天梯赛-练习集)

分而治之,各个击破是兵家常用的策略之一。在战争中,我们希望首先攻下敌方的部分城市,使其剩余的城市变成孤立无援,然后再分头各个击破。为此参谋部提供了若干打击方案。本题就请你编写程序,判断每个方案的可行性。
输入格式:
输入在第一行给出两个正整数 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
using namespace std;
typedef long long int ll;
int f[10010],n,m,x[10010],y[10010],k;
bool used[10010];
void init()
{
 for(int i=1;i<=10010;i++)
 {
  f[i]=i;
 }
}//初始化 
int find(int u)
{
 if(f[u]==u)return u;
 return f[u]=find(f[u]);
}//路径压缩 
void Union(int u,int v)
{
 int a=find(u);
 int b=find(v);
 if(a!=b)
 {
  f[a]=b;
 }
}//并查集 
int count()
{
 int ans=0;
 for(int i=1;i<=n;i++)
 {
  if(f[i]==i)
  {
   ans++;
  }
 }
 return ans;
}//查找图的连通块数 
int main()
{
 cin>>n>>m;
 init();
 for(int i=1;i<=m;i++)
 {
  cin>>x[i]>>y[i];
  Union(x[i],y[i]);
 }
 int c1=count();
 cin>>k;
 while(k--)
 {
  memset(used,false,sizeof(used));//初始化 
  int h,g;
  init();
  cin>>h;
  for(int i=1;i<=h;i++)
  {
   cin>>g;
   used[g]=true;
  }//记录被攻占的城市 
  for(int i=1;i<=m;i++)
  {
   if(used[x[i]]||used[y[i]])continue;//失效的路线 
   Union(x[i],y[i]); 
  }
  c1=count();
  if(c1==n)
  {
   printf("YES\n");
  }
  else printf("NO\n");
 }
 return 0;
}

你可能感兴趣的:(数据结构,图论,算法,数据结构)