0
点赞
收藏
分享

微信扫一扫

*【PAT天梯】分而治之(并查集,暴力)

Brose 2022-06-15 阅读 12

题干:

L2-1 分而治之 (25 分)

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

输入格式:

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

 解题报告:

  因为数据范围不大,考虑O(N*K)的做法:

  对于每个方案使用并查集,最后看是否每个节点的boss是否是自己就可以了。

考虑优化?

AC代码:

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#include<map>
#include<cmath>
#include<cstring>
typedef long long ll;
using namespace std;
const int MAX = 2e5 + 6;
int f[MAX];
int n,m;
int u[MAX],v[MAX];
bool bk[MAX];
void init(int n) {
for(int i = 1; i<=n; i++) f[i] = i;
}
int getf(int v) {
return v == f[v] ? v : f[v] = getf(f[v]);
}
void merge(int u,int v) {
int t1 = getf(u);
int t2 = getf(v);
f[t2] = t1;
}
bool ok() {
for(int i = 1; i<=n; i++) {
if(f[i] != i) return 0 ;
}
return 1;
}
int main()
{
cin>>n>>m;
for(int i = 1; i<=m; i++) {
scanf("%d%d",&u[i],&v[i]);
}
int k;
cin>>k;
for(int i = 1; i<=k; i++) {
init(n);
for(int i = 1; i<=n; i++) bk[i] = 0;
int num;
scanf("%d",&num);
for(int tmp,j = 1; j<=num; j++) {
scanf("%d",&tmp);
bk[tmp]=1;
}
for(int j = 1; j<=m; j++) {
if(bk[u[j]]|| bk[v[j]]) continue;
merge(u[j],v[j]);
}
if(ok()) printf("YES\n");
else printf("NO\n");
}
return 0 ;
}

 


举报

相关推荐

0 条评论