首页 > 代码库 > 生化危机

生化危机

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天感染了城市的数量.
(注意, 每组数据第二行每个数据后边都有一个空格)

23 11 22 33 21 22 3
31 1 121 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;   }}

 

生化危机