当前位置: 代码迷 >> 综合 >> 3-5 分而治之 (15 分)(并查集秒解)
  详细解决方案

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

热度:110   发布时间:2023-11-25 20:02:06.0

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 };}
}