团队程序设计天梯赛 L2-2 病毒溯源

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

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

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

输入格式：

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

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

k 变异株1 …… 变异株k

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

输出格式：

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

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

注：我们称序列 {$ a_1,⋯,a_n $} 比序列 { $b_1,⋯,b_n KaTeX parse error: Expected 'EOF', got '}' at position 1: }̲ “小”，如果存在 1≤k≤n $满足 $a_i=b_i$ 对所有$ i<k$ 成立，且 $a_k<b_k$。

输入样例：

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 <bits/stdc++.h>

using namespace std;

vector<int> mp[10001];
int fa[10001]; //判断一下根节点,既没有做过孩子的就是根节点
vector<int> ans;

void dfs(int x, vector<int> &p)   //如果不加‘&’ 会占用很大内存导致内存超限
{
    if (p.size() > ans.size())
    {
        ans.clear();
        for (int i = 0; i < p.size(); i++)  //更新答案数组
        {
            ans.push_back(p[i]);
        }
    }
    for (int i = 0; i < mp[x].size(); i++)
    {
        p.push_back(mp[x][i]);
        dfs(mp[x][i], p);
        p.pop_back();  //回溯一下
    }
}

int main()
{
    int n;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        int k;
        cin >> k;
        while (k--)
        {
            int x;
            cin >> x;
            mp[i].push_back(x);
            fa[x] = 1;  //标记一下，最后剩下没被标记的就是根节点
        }
        if (mp[i].size())
            sort(mp[i].begin(), mp[i].end());
    }
    for (int j = 0; j < n; j++)
    {
        if (!fa[j])
        {
            vector<int> p;
            p.push_back(j);
            dfs(j, p);
            break;
        }
    }
    cout << ans.size() << endl;
    for (int k = 0; k < ans.size(); k++)
    {
        if (k)
            cout << " ";
        cout << ans[k];
    }
    return 0;
}