分而治之,各个击破是兵家常用的策略之一。在战争中,我们希望首先攻下敌方的部分城市,使其剩余的城市变成孤立无援,然后再分头各个击破。为此参谋部提供了若干打击方案。本题就请你编写程序,判断每个方案的可行性。
输入格式:
输入在第一行给出两个正整数 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
思路:开一个bool数组st,st[i] = true表示第i号城市被摧毁。
对于每一个询问,遍历所有没有被摧毁的城市,判断该城市的所有相邻城市是否被摧毁,若有一个没有被摧毁则输出NO,否者输出YES
#include <bits/stdc++.h> using namespace std; const int N = 1e4 + 10; int n, m, T; // n:城市个数、m:边数、T:T组询问 vector<int> v[N]; // 邻接表 bool st[N]; // st[i]:true表示第i号城市被摧毁 int main(){ // 处理输入 cin >> n >> m; for(int i = 0; i < m; i ++ ) { int a, b; cin >> a >> b; v[a].push_back(b); v[b].push_back(a); // 题意为无向图,要建双向的边 } cin >> T; while(T -- ) { memset(st, 0, sizeof st); // 清空st int k; cin >> k; while(k -- ) { int t; cin >> t; st[t] = true; } bool flag = true; // flag = false 表示某个城市至少有一个相邻城市没有被摧毁 for(int i = 1; i <= n; i ++ ) { if(st[i]) continue; if(flag == false) break; for(int j = 0; j < v[i].size(); j ++ ) { int t = v[i][j]; if(st[t] == false) { flag = false; break; } } } cout << (flag ? "YES" : "NO") << endl; } return 0; }