2021天梯赛L2-2 病毒溯源

L2-2 病毒溯源

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

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

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

输入格式：
输入在第一行中给出一个正整数 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
分析：首先用二维vector将每个病毒产生的变异病毒存起来，然后对每个病毒产生的所有变异病毒从小到大排序。使用dfs从源病毒开始搜索，在递归的过程中每次更新最长链的长度。

#include<bits/stdc++.h>
using namespace std;
vector<int> a[10005];
vector<int> chain;
vector<int> longest;
int flag[10005];
void dfs(int index,vector<int> &chain){
	if(chain.size()>longest.size()){//每次递归都更新longest
		longest.clear();
		longest=chain;
	}
	int j;
	for(j=0;j<a[index].size();j++){ //对a[index]产生的每个变异病毒dfs
		chain.push_back(a[index][j]);//把当前要dfs的病毒放入chain中
		dfs(a[index][j],chain);
		chain.pop_back();//该病毒dfs结束后出chain，下一次循环中的链不能包含这个病毒，因为他们都是index的变异病毒，只能有一个在chain里
	}
}
int main(){
	int n,i;
	cin>>n;
    for(i=0;i<n;i++){
    	int k;
    	cin>>k;
    	while(k--){
    		int x;
    		cin>>x;
    		a[i].push_back(x);//存每个病毒产生的变异病毒
    		flag[x]=1;//这是本题的巧妙之处，怎样找到源病毒？通过把每个病毒产生的变异病毒标志置为1，最后仍为0的就是源病毒
		}
		if(a[i].size()!=0){
			sort(a[i].begin(),a[i].end());//对每个病毒产生的变异病毒从小到大排序
		}
	}
	for(i=0;i<n;i++){
		if(flag[i]==0){
			chain.push_back(i);
			dfs(i,chain);
		}
	}
	cout<<longest.size()<<endl;
	for(i=0;i<longest.size();i++){
		if(i==0) cout<<longest[i];
		else cout<<" "<<longest[i];
	}
	return 0;
}