ZZULIOJ 1787 生化危机（vector+DFS）

1787: 生化危机

Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 64 Solved: 19

SubmitStatusWeb Board

Description

X博士想造福人类, 研发一种可以再生肢体的药物, 可是很不幸......研究失败了, 他在蜥蜴身上实验的时候, 蜥蜴发生了变异, 更糟糕的是, 蜥蜴逃出了生化实验室.

恐怖的事情发生了, 疫情以X博士所在的城市为中心向四周扩散开, 最终, 整个地球上的城市都被感染了.假设整个地球一共有N个城市, 这N个城市是连通的, 有N-1条通道把他们连接起来.病毒会以一座城市为中心,在一天的时间内, 会把和他相连的周围的所有城市感染. 那么, 多少天的时间, 整个地球的城市都感染呢?

Input

第一行输入一个T(T <= 50), 表示一共有T组测试数据. 每组数据第一行两个数n, k. n表示有n(2 <= n <= 10000)个城市, 代表城市1-n, k是X博士所在的城市. 接下来n-1行, 每行有两个数u, v, 表示城市u和v之间有一条通道.

Output

每组数据第一行输出第一个数 x , 表示整个地球感染需要x天. 接下来 x 个数, 第i个数表示第i天感染了城市的数量. (注意，每组数据第二行每个数据后边都有一个空格）.

Sample Input

23 11 22 33 21 22 3

Sample Output

31 1 121 2

思路：用vector存储相连的结点，然后从起点开始搜索，对感染到的结点进行标记，然后在搜索过程中进行计数，DFS几次就是第几天+1
ac代码：

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<vector>
#include<iostream>
#include<algorithm>
#define fab(a) (a)>0?(a):(-a)
#define LL long long
#define MAXN 100010
#define mem(x) memset(x,0,sizeof(x))
#define INF 0xfffffff
#define PI acos(-1.0)
using namespace std;
vector<int>map[MAXN];
int bz[MAXN];
int ans[MAXN];
void dfs(int v,int x)
{for(int i=0;i<map[v].size();i++){if(bz[map[v][i]]==0){bz[map[v][i]]=1;ans[x]++;dfs(map[v][i],x+1);}}
}
int main()
{int t,i;int n,m,q,w;scanf("%d",&t);while(t--){scanf("%d%d",&n,&m);mem(ans);mem(bz);for(i=1;i<=n;i++){ map[i].clear();} for(i=1;i<n;i++){scanf("%d%d",&q,&w);map[q].push_back(w);map[w].push_back(q);}bz[m]=1;dfs(m,2);int cnt=1;for(i=1;i<=n;i++) if(ans[i]) cnt++;printf("%d\n",cnt);printf("1 ");for(i=1;i<=n;i++) if(ans[i]) printf("%d ",ans[i]);printf("\n");}return 0;
}