【PTA】L2-038 病毒溯源 (25 分)（DFS求最长路径并记录）

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

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

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

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

经典的DFS题，ve代表当前分支的序列，ans记录最长且最小的序列，每次在递归入口进行更新。
此题如果对每个点进行搜索，有一个测试点会超时。我试过记忆化搜索，内存会超限。
再看样例，每个点的后面是源头的后代，则说明后代的长度不可能比源头的长度还长，所以可以把后代标记起来不再搜索。

#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
const int N=1e4+10;
vector<int>v[N],ve,ans;
int vis[N],v1[N];
void dfs(int p,int root){
	ve.push_back(p);
	vis[p]=1;
	if(ve.size()>ans.size()){
		ans=ve;
	}
	else if(ve.size()==ans.size()){
		int f=0;
		for(int i=0;i<ve.size();i++){
			if(ve[i]>ans[i]){
				f=1;
				break;
			}
			else if(ve[i]<ans[i]){
				break;
			}
		}
		if(f==0){
			ans=ve;
		}
	}
	if(v1[root]&&v1[root]+ve.size()<ans.size()) return ;
	for(int i=0;i<v[p].size();i++){
		int to=v[p][i];
		if(vis[to]==0){
			dfs(to,root);
		}
	}
	vis[p]=0;
	ve.pop_back();
}
int main() {
	int n,st,k,x;
	scanf("%d",&n);
	for(int i=0;i<n;i++){
		scanf("%d",&k);
		while(k--){
			scanf("%d",&x);
			v[i].push_back(x);
			v1[x]=1;
		}
	}
	for(int i=0;i<n;i++){
		if(v1[i]==0)
			dfs(i,i);
	}
	printf("%d\n",ans.size());
	for(int i=0;i<ans.size();i++){
		if(i) printf(" ");
		printf("%d",ans[i]);
	}
	return 0;
}