L2-025 分而治之(图)
(这不会是我最后一天写算法题的博客吧。。。有点感伤。。。)
题目:
分而治之,各个击破是兵家常用的策略之一。在战争中,我们希望首先攻下敌方的部分城市,使其剩余的城市变成孤立无援,然后再分头各个击破。为此参谋部提供了若干打击方案。本题就请你编写程序,判断每个方案的可行性。
输入格式:
输入在第一行给出两个正整数 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
思路:
用一个vector二维数组存储与每座城市相通的城市,一个数组存储与每座城市相通的城市数。每毁灭一座城市就将该城市相邻的城市数变为0,并且将之前与其相邻的城市的相邻城市数减一。最后通过遍历这个数组判断是否每座城市相通的城市数都为0。当然,每判断一次数组都要还原,所以要定义两个数组,一个用来给另一个还原。
上代码:
#include <iostream> #include <vector> using namespace std; int a[10005],a1[10005]; void check(){ for(int i=0;i<10005;i++){ if(a1[i]>0){ printf("NO\n"); return ; } } printf("YES\n"); return ; } int main() { int n,m,c1,c2,n1,m1,x; cin>>n>>m; vector<vector<int>> v(n+1); while(m--){ cin>>c1>>c2; v[c1].push_back(c2); v[c2].push_back(c1); a[c1]++; a[c2]++; } cin>>n1; while(n1--){ cin>>m1; for(int i=1;i<=n;i++) a1[i]=a[i]; while(m1--){ cin>>x; a1[x]=0; for(int j=0;j<v[x].size();j++) a1[v[x][j]]--; } check(); } return 0; }