描述
X博士想造福人类, 研发一种可以再生肢体的药物, 可是很不幸......研究失败了, 他在蜥蜴身上实验的时候, 蜥蜴发生了变异, 更糟糕的是, 蜥蜴逃出了生化实验室.
恐 怖的事情发生了, 疫情以X博士所在的城市为中心向四周扩散开, 最终, 整个地球上的城市都被感染了.假设整个地球一共有N个城市, 这N个城市是连通的, 有N-1条通道把他们连接起来.病毒会以一座城市为中心,在一天的时间内, 会把和他相连的周围的所有城市感染. 那么, 多少天的时间, 整个地球的城市都感染呢?
输入
第一行输入一个T(T <= 50), 表示一共有T组测试数据.
每组数据第一行两个数n, k. n表示有n(2 <= n <= 10000)个城市, 代表城市1-n, k是X博士所在的城市.
接下来n-1行, 每行有两个数u, v, 表示城市u和v之间有一条通道.
输出
每组数据第一行输出第一个数 x , 表示整个地球感染需要x天.
接下来 x 个数, 第i个数表示第i天感染了城市的数量.
(注意, 每组数据第二行每个数据后边都有一个空格)
样例输入1
2 3 1 1 2 2 3 3 2 1 2 2 3
样例输出1
3 1 1 1 2 1 2 这是一道搜索题,不断搜索相邻的边,标记,并记录是第几波,记录这一波的数量,最初用bfs超时了,毕竟搜起来有些无脑,然后突然想起最近看的啊哈算法上说的邻接表(一次也没用过),看起来挺牛逼,不如试试,答案不对,想了想,应该双向记录,然后总是runtime error,因为我把一句用到队列顶部元素的语句放在,出队后面,估计是用了已经释放的空间,改了改就过了。 代码:
#include <iostream> #include <cstring> #include <cstdio> #include <queue> using namespace std; int n,k,c; int u[20005],v[20005],first[20005],nexti[20005]; struct que { int va,d; }temp; int main() { int T; queue<que>q; int cou[10005],vis[10005]; cin>>T; while(T--) { scanf("%d%d",&n,&k); memset(first,-1,sizeof(first)); memset(nexti,-1,sizeof(nexti)); memset(cou,0,sizeof(cou)); memset(vis,0,sizeof(vis)); for(int i=0;i<n-1;i++) { cin>>u[i]>>v[i]; } for(int i=n-1;i<2*(n-1);i++) { u[i]=v[i-n+1]; v[i]=u[i-n+1]; } for(int i=0;i<(n-1)*2;i++) { nexti[i]=first[u[i]],first[u[i]]=i; } vis[k]=1; temp.va=k,temp.d=1; q.push(temp); c=1; cou[c]++; while(!q.empty()) { k=first[q.front().va]; while(k!=-1) { if(!vis[v[k]]) { vis[v[k]]=1; temp.va=v[k],temp.d=q.front().d+1; c=temp.d; cou[c]++; q.push(temp); } k=nexti[k]; } q.pop(); } cout<<c<<endl; for(int i=1;i<=c;i++) cout<<cou[i]<<‘ ‘; cout<<endl; } }
时间: 2024-10-10 00:17:01