邻接表bfs练习 病毒溯源

题目描述

病毒容易发生变异。某种病毒可以通过突变产生若干变异的毒株，而这些变异的病毒又可能被诱发突变产生第二代变异，如此继续不断变化。

现给定一些病毒之间的变异关系，要求你找出其中最长的一条变异链。

在此假设给出的变异都是由突变引起的，不考虑复杂的基因重组变异问题 —— 即每一种病毒都是由唯一的一种病毒突变而来，并且不存在循环变异的情况。

输入格式：

输入在第一行中给出一个正整数 N（≤104），即病毒种类的总数。于是我们将所有病毒从 0 到 N−1 进行编号。

随后 N 行，每行按以下格式描述一种病毒的变异情况：

k 变异株1 …… 变异株k

其中 k 是该病毒产生的变异毒株的种类数，后面跟着每种变异株的编号。第 i 行对应编号为 i 的病毒（0≤i<N）。题目保证病毒源头有且仅有一个。

输出格式：

首先输出从源头开始最长变异链的长度。

在第二行中输出从源头开始最长的一条变异链，编号间以 1 个空格分隔，行首尾不得有多余空格。如果最长链不唯一，则输出最小序列。

注：我们称序列 { a1​,⋯,an​ } 比序列 { b1​,⋯,bn​ } “小”，如果存在 1≤k≤n 满足 ai​=bi​ 对所有 i<k 成立，且 ak​<bk​。

输入样例：

10
3 6 4 8
0
0
0
2 5 9
0
1 7
1 2
0
2 3 1

输出样例：

4
0 4 9 1

 分析

每一个序号可以看成一个父节点，他能变成的就是他的字节点（但因为会往别处变异），所以后面画出来是一个有向无环图（因为有环之后那就可以循环变异，最大长度即为无限）

例如给出的数据可以画出图

第一个数据代表共有0-9 10个种类的病毒，下面代表 第i种有几个子节点，3 6 4 8 代表第0个病毒有6 4 8 三个子节点

随后找到第一个根节点，开始广度搜索，维护一个长度数组，记录最深处的节点。

代码如下

#include<bits/stdc++.h>

using namespace std;

const int N = 1e5+10;

vector<int> graph[N];//邻接表

int parent[N];//父节点

int deepth[N];//深度

int maxdeepth = -1,maxnode;//最深节点默认为0，以防仅有一个0节点；

int n;

void bfs(int root)//正常bfs模版
{
    queue<int> que;
    que.push(root);
    deepth[root] = 1;
    while(!que.empty())
    {
        int x = que.front();
        que.pop();
        for(int i = 0;i<graph[x].size();i++)
        {
            parent[graph[x][i]] = x;
            deepth[graph[x][i]] = deepth[x] +1;
            if(deepth[graph[x][i]] > maxdeepth)
            {
                maxdeepth = deepth[graph[x][i]];
                maxnode = graph[x][i];
            }
            que.push(graph[x][i]);
        }
    }
}

void print(int maxnode)//从最深节点倒着找到父节点
{
    vector<int> arr;

    int x = maxnode;
    arr.push_back(x);
    while(parent[x] != -1)
    {
        x = parent[x];
        arr.push_back(x);
    }
    reverse(arr.begin(),arr.end());//翻转

    for(int i = 0;i<arr.size();i++)
    {
        if(i == arr.size()-1)
            cout<<arr[i];
        else
            cout<<arr[i]<<' ';
    }
}

int main()
{
    cin>>n;
    memset(parent,-1,sizeof(parent));
    for(int i = 0;i<n;i++)
    {
        int x = 0;
        cin>>x;
        for(int j = 0;j<x;j++)
        {
            int r = 0;
            cin>>r;
            graph[i].push_back(r);
            parent[r] = i;
        }
        sort(graph[i].begin(),graph[i].end());//*****一个sort即可解决按照字典序最小输出；
    }
    for(int i = 0;i<n;i++)
    {
        if(parent[i] == -1)
            bfs(i);
    }

    cout<<deepth[maxnode]<<endl;

    print(maxnode);
    
    return 0;
}