L2-025 分而治之 (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
code:
#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
int countcity[10010];
int countcc[10010];
int n,m;
void check(){
for(int i=0;i<n;i++){
if(countcc[i]>0){//剩余相邻的城市个数>0
cout<<"NO";
return;
}
}
cout<<"YES";
return;
}
int main(){
cin>>n>>m;
vector<vector<int>> v(n+1);
for(int i=0;i<m;i++){//构建地图
int a,b;
cin>>a>>b;
countcity[a]++;
countcity[b]++;
v[a].push_back(b);
v[b].push_back(a);
}
int k;
cin>>k;
for(int i=0;i<k;i++){
int np;
cin>>np;
for(int j=1;j<=n;j++){
countcc[j]=countcity[j];
}
for(int j=0;j<np;j++){
int t;
cin>>t;
countcc[t]=0;//被攻占
for(int p=0;p<v[t].size();p++){
countcc[v[t][p]]--;//兄弟城市的相邻城市数-1
}
}
check();
if(i!=k-1)
cout<<endl;
}
}
用二维数组保存每个城市所相邻的城市,并用countcity数组来保存 个数。
对于每一种攻占策略,用countcc保存当前策略目前各个城市所相邻的城市个数,每攻占一个城市,其所相邻的城市对应的countcc-1。
最后利用check检查是否有城市的相邻个数还大于0,如果没有则符合。