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

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

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

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

输入格式：
输入在第一行中给出一个正整数 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

分析：Edge[i]中保存第i种病毒可能产生的变异病毒编号，in[i]中保存编号i的入度，入度为零的病毒为源头病毒，pa[i]中保存病毒由哪种病毒变异而来（上级病毒）。为了保证序列最小，在输入Edge[i]后可对其进行一次排序～然后通过BFS找到变异次数，过程中由Long和ans标记最多变异次数量以及对应的病毒编号，最后通过pa找回完整变异链～

#include <bits/stdc++.h>
using namespace std;
#define pii pair<int ,int>
int n, k, t, S, Long = 0,  ans, in[10005], pa[10005];
vector<int> F, Edge[10005];
queue<pii> Q;
int main() {
    cin >> n;
    for (int i = 0; i < n; i++) {
        cin >> k;
        for (int j = 0; j < k; j++) {
            cin >> t;
            in[t]++;
            pa[t] = i;
            Edge[i].push_back(t);
        }
        sort(Edge[i].begin(), Edge[i].end());
    }
    for (int i = 0; i < n; i++) if (in[i] == 0) S = i;
    while(!Q.empty()) Q.pop();
    Q.push({S, 1});
    while (!Q.empty()) {
        int now = Q.front().first, D = Q.front().second;
        Q.pop();
        if (D > Long) {
            Long = D;
            ans = now;
        }
        for (auto nex : Edge[now]) Q.push({nex, D + 1});
    }
    cout << Long << '\n';
    while (ans != S) {
        F.push_back(ans);
        ans = pa[ans];
    }
    F.push_back(S);
    for (int i = F.size() - 1; ~i; --i) {
        cout << F[i];
        if (i != 0) cout << ' ';
    }
    return 0;
}

#include <bits/stdc++.h>

using namespace std;

#define pii pair<int ,int>

int n, k, t, S, Long = 0, ans, in[10005], pa[10005];

vector<int> F, Edge[10005];

queue<pii> Q;

int main() {

cin >> n;

for (int i = 0; i < n; i++) {

cin >> k;

for (int j = 0; j < k; j++) {

cin >> t;

in[t]++;

pa[t] = i;

Edge[i].push_back(t);

}

sort(Edge[i].begin(), Edge[i].end());

}

for (int i = 0; i < n; i++) if (in[i] == 0) S = i;

while(!Q.empty()) Q.pop();

Q.push({S, 1});

while (!Q.empty()) {

int now = Q.front().first, D = Q.front().second;

Q.pop();

if (D > Long) {

Long = D;

ans = now;

}

for (auto nex : Edge[now]) Q.push({nex, D + 1});

}

cout << Long << '\n';

while (ans != S) {

F.push_back(ans);

ans = pa[ans];

}

F.push_back(S);

for (int i = F.size() - 1; ~i; --i) {

cout << F[i];

if (i != 0) cout << ' ';

}

return 0;

}

❤ 点击这里 -> 订阅《PAT | 蓝桥 | LeetCode学习路径 & 刷题经验》by 柳婼

❤ 点击这里 -> 订阅《从放弃C语言到使用C++刷算法的简明教程》by 柳婼

❤ 点击这里 -> 订阅PAT甲级乙级、蓝桥杯、GPLT天梯赛、LeetCode题解离线版