Java算法题目——病毒溯源

145 阅读2分钟

这是我参与11月更文挑战的第20天,活动详情查看:2021最后一次更文挑战

病毒溯源

病毒容易发生变异。

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

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

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

输入格式

输入在第一行中给出一个正整数 NN,即病毒种类的总数。于是我们将所有病毒从 00 到 N−1N−1 进行编号。

随后 NN 行,每行按以下格式描述一种病毒的变异情况:

k 变异株1 …… 变异株k

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

输出格式

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

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

注:我们称序列 {a1,…,an}{a1,…,an} 比序列 {b1,…,bn}{b1,…,bn} “小”,如果存在 1≤k≤n1≤k≤n 满足 ai=biai=bi 对所有 i<ki<k 成立,且 ak<bkak<bk。

数据范围

1≤N≤1041≤N≤104

输入样例:

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

代码



import java.lang.reflect.Array;
import java.util.Arrays;
import java.util.Scanner;

public class Main {

	final static int N=10010,M=10010;
	
	int n;
	static int h[]=new int[N],
		e[]=new int[N],
		ne[]=new int[M],
		idx;
	static int son[]=new int[N];
	static boolean st[]=new boolean[N];
	
	public static void main(String[] args) {
		
		Scanner sc=new Scanner(System.in);
		
		Arrays.fill(h,-1);
		
		int n=sc.nextInt();
		
		for(int i=0;i<n;i++) {
			int count=sc.nextInt();
			while(count!=0) {
				int x=sc.nextInt();
				add(i,x);
				count--;
				st[x]=true;
			}
		}
		int root=0;
		while(st[root]) {
			root++;
		}
		
		System.out.println(dfs(root));
		System.out.print(root);
		while(son[root]!=-1) {
			root=son[root];
			System.out.print(" "+root);
		}
		
	}

	static void add(int a,int b) {
		e[idx]=b;
		ne[idx]=h[a];
		h[a]=idx++;
	}
	
	static int dfs(int u) {
		
		int res=0;
		son[u]=-1;
		for(int i=h[u];i!=-1;i=ne[i]) {
			int j=e[i];
			int d=dfs(j);
			if(res<d) {
				res=d;
				son[u]=j;
			}else if(res==d) {
				son[u]=Math.min(son[u],j);
			}
		}
		return res+1;
	}
	
}