团体程序设计天梯赛-练习集-L2-038 病毒溯源 (25 分)

 

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

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

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

输入格式：

输入在第一行中给出一个正整数 N（≤10^4），即病毒种类的总数。于是我们将所有病毒从 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

代码示例：

#include <iostream>
#include <vector>
using namespace std;

const int N = 10000;   //病毒种类总数的最大值
int n;                 //病毒种类的总数
vector<int> v[N];      //存下初始病毒关系
vector<int> ans, tmp;  //ans为不包含根病毒的最长变异链，tmp用于暂存ans
bool vis[N];           //标记数组，true为发生变异，用于查找根病毒
int cnt = 0;           //记录变异链长度sum值

void DFS(int u, int sum) {  //sum为变异链长度
    if (sum > cnt) {        //更新记录ans和cnt值
        ans = tmp;
        cnt = sum;
    } else if (sum == cnt && tmp < ans) {  //sum==cnt时，输出最小序列
        ans = tmp;
    }
    for (unsigned int i = 0; i < v[u].size(); i++) {  //v[u].size()为无符号数
        tmp.push_back(v[u][i]);                       //不包括根病毒的变异链
        DFS(v[u][i], sum + 1);
        tmp.pop_back();  //弹出回溯
    }
}

int main() {
    cin >> n;
    for (int i = 0; i < n; i++) {
        int k;  //该病毒产生的变异毒株的种类数
        cin >> k;
        while (k--) {
            int x;
            cin >> x;
            v[i].push_back(x);
            vis[x] = true;  //不是根病毒
        }
    }
    int root = 0;  //根病毒指针
    while (vis[root]) {
        root++;
    }
    DFS(root, 1);         //从根病毒开始深度优先遍历找到链长最长的变异链，初始变异链长为1
    cout << cnt << endl;  //最长变异链的长度
    cout << root;         //根病毒
    for (auto i : ans) {  //遍历输出ans
        cout << " " << i;
    }
    return 0;
}