分而治之,各个击破是兵家常用的策略之一。在战争中,我们希望首先攻下敌方的部分城市,使其剩余的城市变成孤立无援,然后再分头各个击破。为此参谋部提供了若干打击方案。本题就请你编写程序,判断每个方案的可行性。
输入格式:
输入在第一行给出两个正整数 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
解决方案
使用set类型存放方案
考虑图的连通性问题,用vector类型存放每个节点之间的关系,可以解决内存问题,define MAX的时候要注意,MAX = 10000会出现段错误问题,需要更大。
主要思路是,考虑每个不被攻占的节点除了已被攻占的节点外,是否存在其他的联通,若存在就无法达到分而治之的目的。若只有已被攻占的节点,那么就是已经成功把所有的节点分隔开。
解决代码
#include <bits/stdc++.h>
#define MAX 10000
using namespace std;
int n, m;
int np;
int planCount;
set<int> s;
bool flag = true;
void check(vector<int> g[])
{
for (int i = 0; i < n; i++)
{
/* code */
if (!s.count(i))
{
for (int j = 0; j < g[i].size(); j++)
{
/* code */
if (!s.count(g[i][j]))
{
// cout << j << ":" << g[i][j] << "\t";
flag = false;
return;
}
}
}
}
}
int main()
{
vector<int> g[MAX];
cin >> n >> m;
// build graph
for (int i = 0; i < m; i++)
{
int pos_01, pos_02;
cin >> pos_01 >> pos_02;
g[pos_01].push_back(pos_02);
g[pos_02].push_back(pos_01);
}
cin >> planCount;
for (int i = 0; i < planCount; i++)
{
/* code */
s.clear();
flag = true;
int size;
cin >> size;
for (int k = 0; k < size; k++)
{
cin >> np;
s.insert(np);
}
check(g);
if (flag)
{
cout << "Yes\n";
}
else
{
cout << "No\n";
}
}
}