PTA天梯赛L2-038 病毒溯源

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

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

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

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

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

k 变异株1 …… 变异株k

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

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

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

注：我们称序列 { a​1​​ ,⋯,a​n​​ } 比序列 { b​1​​ ,⋯,b​n​​ } “小”，如果存在 1≤k≤n 满足 a​i​​ =b​i​​ 对所有 i<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

思路：

病毒通过变异产生新病毒，若将病毒看作一个结点，则a病毒可变异为b病毒——可以理解为结点a到结点b中有一条有向边。

由于不存在循环变异，所以这是一条有向无环图。

所以这是单源最短路！Dijkstra！SPFA！拓扑！

事实上这只是个dfs搜索题，将每条路都搜索到底并记录长度和路径，然后对比最长路径和当前路径长度，若当前路径更长则更新ans数组和最长路长度maxlen。

至于最小序列只需要在建图时先存结点小的边即可，这样在dfs遍历的时候永远都是先遍历到结点较小的边。

下附ac代码

#include<queue>
#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;

//当前已知最长路长度
int maxlen;	

//temp是用于暂存路径的数组，ans为最终搜索到的最长路
int temp[100005],ans[100005];

//in数组记录每个点的入度，用于查找源点
int in[100005];	 

//End数组用于记录该点的是否为终点
bool End[100005];

//前向星建图
int cur,head[100005];
struct Edge{
	int nxt,to;
}edge[100005];
void build(int u,int v){
	edge[++cur].nxt=head[u];
	edge[cur].to=v;
	head[u]=cur;
}

//递归搜索每条道路
void dfs(int len,int e){//len代表当前路径长，e为当前所在节点
	if(End[e]){	//当前结点已经是某条路径的终点
		if(maxlen<len){	//若最大路径长度小于当前路径，更新
			maxlen=len;
			for(int i=0;i<len;++i)//更新ans数组
				ans[i]=temp[i];
		}
		return ;
	}

	for(int i=head[e];i;i=edge[i].nxt){
		temp[len-1]=edge[i].to;	//先将路径暂存在temp数组中
		dfs(len+1,edge[i].to);	//递归查找下一个结点
	}
}

int main()
{
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	int n,m,k,a[10005],s;
	cin>>n;
	for(int i=0;i<n;++i){
		cin>>k;

		//k为0说明该病毒不能变异成别的什么了，即为路径终点
		if(k==0)	//变更End状态为true
			End[i]=true;
		for(int j=0;j<k;++j)
			cin>>a[j];
		
		//由于是前向星建图所以要先将结点从大到小排序后建图才能做到先搜结点小的边
		//greater<int>()让sort可以从大到小排序而非从小到大
		sort(a,a+k,greater<int>());

		for(int j=0;j<k;++j){
			++in[a[j]];	//入度自增一
			build(i,a[j]);//前向星建图
		}
	}
	for(int i=0;i<n;++i){
		if(!in[i]){//说明没有父节点，该点为源点
			s=i;
			break;
		}//由于题目说了只有一个源点，所以找到一个源点即可跳出循环
	}
	dfs(1,s);//初始长度为1，源点是s
	cout<<maxlen<<endl;

	//由于在递归过程中并不会经过源点，所以ans数组中是没有源点的，所以在这里直接输出源点
	cout<<s;
	for(int i=1;i<maxlen;++i){//格式控制，防止最后有多余的空格
		cout<<" "<<ans[i-1];
	}
}