L2-038 病毒溯源

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

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

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

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

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

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

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

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

输入样例：

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

代码

主要思路看代码注释，用的是bfs

#include<bits/stdc++.h>
using namespace std;
vector<int>G[10005]; //存图
int maxx,a[10005],s,j;//数组a记录变异毒株，用来找出源头，s是源头，maxx是最长变异长度，j记录最长变异长度的最后一个节点
map<int,int>m; //记录每个变异毒株的对应的病毒，因为每一种病毒都是由唯一的一种病毒突变，可以由最后一个节点来推出整个变异链
struct node
{
    int step;
    int sum;
};
int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        int k,x;
        cin>>k;
        while(k--)
        {
            cin>>x;
            a[x]++;
            G[i].push_back(x); //存图
            m[x]=i; //记录每个变异毒株的对应的病毒，且是唯一的
        }
    }
    //将存图的节点由大到小排序，因为当长度一样的时候，要输出序列小的，我用的是bfs,这样排序后，后面出队的就是序列小的，解决了这个序列问题
    for(int i=0;i<n;i++) sort(G[i].begin(),G[i].end(),greater<int>());
    for(int i=0;i<n;i++) //找源头
    {
        if(a[i]==0)
        {
            s=i;
            break;
        }
    }
    queue<node>q;
    q.push({s,1});
    while(q.size())
    {
        node f=q.front();
        q.pop();
        if(f.sum>=maxx)
        {
            j=f.step;
            maxx=f.sum;
            
        }
        for(auto i:G[f.step])
        {
            q.push({i,f.sum+1});
        }
    }
   cout<<maxx<<"\n";
    if(maxx==1)  //变异链最长是1的时候，也就是只有源头,这个情况是一个点
    {
        cout<<s;
        return 0;
    }
   stack<int>st; //用来存变异链的节点
   st.push(j); //最后一个节点
   while(m[j]!=s) //由最后一个节点推出全部节点，存入st栈
   {
       st.push(m[j]);
       j=m[j];
   }
   cout<<s;  //输出
    while(st.size())
    {
        cout<<" "<<st.top();
        st.pop();
    }
}