编写一个C++代码分而治之各个击破是兵家常用的策略之一。在战争中我们希望首先攻下敌方的部分城市使其剩余的城市变成孤立无援然后再分头各个击破。为此参谋部提供了若干打击方案。本题就请你编写程序判断每个方案的可行性。输入格式:输入在第一行给出两个正整数 N 和 M均不超过10 000分别为敌方城市个数于是默认城市从 1 到 N 编号和连接两城市的通路条数。随后 M 行每行给出一条通路所连接的两个城市的编
首先,我们可以使用邻接表来表示城市之间的通路关系。具体实现如下:
#include <iostream>
#include <vector>
using namespace std;
// 邻接表表示城市之间的通路关系
vector<vector<int>> graph;
// 标记城市是否被攻下
vector<bool> captured;
// 深度优先搜索判断方案的可行性
bool dfs(int city) {
// 如果城市已经被攻下,则返回false
if (captured[city]) {
return false;
}
// 标记城市为已攻下
captured[city] = true;
// 遍历与该城市相连的城市
for (int i = 0; i < graph[city].size(); i++) {
int nextCity = graph[city][i];
// 如果相连的城市已经被攻下,则返回false
if (captured[nextCity]) {
return false;
}
// 继续深度优先搜索
if (!dfs(nextCity)) {
return false;
}
}
return true;
}
int main() {
int N, M;
cin >> N >> M;
// 初始化邻接表和标记数组
graph.resize(N + 1);
captured.resize(N + 1, false);
// 读取通路信息
for (int i = 0; i < M; i++) {
int city1, city2;
cin >> city1 >> city2;
// 添加通路
graph[city1].push_back(city2);
graph[city2].push_back(city1);
}
int K;
cin >> K;
// 遍历每个方案
for (int i = 0; i < K; i++) {
int Np;
cin >> Np;
// 初始化标记数组
fill(captured.begin(), captured.end(), false);
// 读取攻下的城市编号
for (int j = 0; j < Np; j++) {
int city;
cin >> city;
// 标记城市为已攻下
captured[city] = true;
}
// 判断方案的可行性
bool feasible = true;
for (int j = 1; j <= N; j++) {
if (!dfs(j)) {
feasible = false;
break;
}
}
// 输出结果
if (feasible) {
cout << "YES" << endl;
} else {
cout << "NO" << endl;
}
}
return 0;
}
该代码首先使用邻接表表示城市之间的通路关系,然后使用深度优先搜索来判断每个方案的可行性。具体步骤如下:
- 读取输入的城市数量 N 和通路数量 M;
- 初始化邻接表和标记数组;
- 读取每条通路的信息,并添加到邻接表中;
- 读取方案数量 K;
- 对于每个方案,首先读取攻下的城市数量 Np,并初始化标记数组;
- 读取攻下的城市编号,并将其标记为已攻下;
- 对于每个城市,如果该城市未被攻下,则进行深度优先搜索;
- 如果存在未被攻下的城市,则该方案不可行,输出NO;否则输出YES;
- 返回第 5 步,继续处理下一个方案;
- 结束。
该代码的时间复杂度为 O(K * (N + M)),其中 K 为方案数量,N 为城市数量,M 为通路数量
原文地址: https://www.cveoy.top/t/topic/h6wd 著作权归作者所有。请勿转载和采集!