天梯赛L2-025 分而治之

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，如果没有则符合。