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;
}