L2-038 病毒溯源

题目

L2-038 病毒溯源

分数 25

作者 陈越

单位 浙江大学

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

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

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

输入格式：

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

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

栈限制

8192 KB

思路

很显然这道题是一个深度搜索（dfs）问题，要找到子节点的最深链条。

难点如下：

一、如何找到根节点

根节点题目保证只有一个，那么只需要找到入度为0的节点即是根节点。

那么就转换为找入度为0的节点，即排除入度非零的节点。

初始化一个子节点标记数组，初始化为全false。在输入节点的子节点的过程中，这些子节点入度大于0，标记为是个子节点，该节点标记赋值为true。全部录入后，子节点标记仍然为false的节点是根节点。

二、如何深度搜索

在深度搜索的过程中需要记录最长子链，需要用到回溯算法。

由于该题题目性质，一个节点只会搜索到一次，因此无需有访问标志指示节点是否被访问过。

回溯算法核心是：

1、节点放入当前链条中。

2、深度搜索节点的子节点。

3、节点与所有子节点遍历后，节点弹出当前链条。

三、如何找到最小序列

很简单，只需要将输入的每个节点的子节点进行排序，并且在相同链条长度的情况下不更新找到的链条即可。

代码

#include <iostream>
using namespace std;

vector<int> son[10001];
vector<int> len;  // 子序列长度
bool isson[10001]; // 是否是个子节点
vector<int> out; // 找到的最长链

void dfs(int i, vector<int> &now) {
	if (now.size() > out.size()) {
		out = now;
	}
	for (int ii = 0; ii < len[i]; ii++) {
		now.push_back(son[i][ii]);
		dfs(son[i][ii], now);
		now.pop_back();
	}
}

int main() {
	int n;
	cin >> n;
	for (int i = 0; i < n; i++) {
		int k;
		cin >> k;
		len.push_back(k);
		while (k--) {
			int t;
			cin >> t;
			isson[t] = true;
			son[i].push_back(t);
		}
		if (son[i].size() > 0) {
			sort(son[i].begin(), son[i].end());
		}
	}
	for (int i = 0; i < n; i++) {
		if (!isson[i]) {
			vector<int> now; // 当前遍历的链条
			now.push_back(i);
			dfs(i, now);
		}
	}
	cout << out.size() << endl;
	for (int i = 0; i < out.size(); i++) {
		if (!i)
			cout << out[i];
		else
			cout << " " << out[i];
	}
}