题目
病毒容易发生变异。某种病毒可以通过突变产生若干变异的毒株,而这些变异的病毒又可能被诱发突变产生第二代变异,如此继续不断变化。
现给定一些病毒之间的变异关系,要求你找出其中最长的一条变异链。
在此假设给出的变异都是由突变引起的,不考虑复杂的基因重组变异问题 —— 即每一种病毒都是由唯一的一种病毒突变而来,并且不存在循环变异的情况。
输入格式:
输入在第一行中给出一个正整数 N(≤104 ),即病毒种类的总数。于是我们将所有病毒从 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
数据范围
1≤N≤104
题解
PTA平台上java过不了,先放一个c++的。
#include <iostream>
#include <vector>
#include <cstring>
using namespace std;
const int N = 10010;
const int M = N;
int h[N];
int e[N];
int ne[M];
int son[N];
int st[N];
int idx;
int n;
int max_val = -1;
int pos;
void add(int a, int b) {
e[idx] = b;
ne[idx] = h[a];
h[a] = idx++;
}
int dfs(int root) {
int res = 0;
son[root] = -1;
for (int i = h[root]; i != -1; i = ne[i]) {
int j = e[i];
int t = dfs(j);
if (t > res) {
res = t;
son[root] = j;
}
else if (res == t) {
son[root] = min(son[root], j);
}
}
return res + 1;
}
int main() {
cin >> n;
memset(h, -1, sizeof(h));
for (int i = 0; i < n; i++) {
int t;
cin >> t;
for (int j = 0; j < t; j++) {
int tt;
cin >> tt;
add(i, tt);
st[tt] = 1;
}
}
for (int i = 0; i < n; i++) {
if (st[i] == 0) {
int t = dfs(i);
if (t > max_val) {
max_val = t;
pos = i;
}
}
}
cout << max_val << endl;
cout << pos;
while (son[pos] != -1) {
cout << " " << son[pos];
pos = son[pos];
}
return 0;
}
Acwing上可过 java代码
import java.util.Arrays;
import java.util.Scanner;
/**
* @author akuya
* @create 2024-04-04-19:10
*/
public class Main {
static int N=10010;
static int n;
static int idx;
static int h[]=new int[N];
static int ne[]=new int[N];
static int e[]=new int[N];
static int son[]=new int[N];
static int st[]=new int[N];
public static void main(String[] args) {
Scanner scanner=new Scanner(System.in);
n=scanner.nextInt();
Arrays.fill(h,-1);
for(int i=0;i<n;i++){
int t=scanner.nextInt();
while(t--!=0){
int tt=scanner.nextInt();
add(i,tt);
st[tt]=1;
}
}
//获取树根
int tt=0;
while(st[tt]!=0)tt++;
int num=dfs(tt);
System.out.println(num);
System.out.print(tt+" ");
for(int i=0;i<num-1;i++){
System.out.print(son[tt]+" ");
tt=son[tt];
}
}
public static void add(int a,int b){
e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}
public static int dfs(int t){
int res=0;
for(int i=h[t];i!=-1;i=ne[i]){
int j=e[i];
int d=dfs(j);
if(d>res||d==res&&j<son[t]){
res=d;
son[t]=j;
}
}
return res+1;
}
}
思路
——本题考察树的构建以及树的遍历,本题解使用静态链表对树进行存贮,使用递归的思想配合深搜算出最大深度,通过创建数组son来储存最长病毒链。本题考察的比较综合,要掌握好树的同时能够结合其他算法知识点,有一点笑笑的难度。
——补充:本题有几个细节,首先需要找到树的头结点。因此我们用到st数组,在构造树的时候顺便遍历谁没有父节点,那么最后遍历st即可知谁是父节点。
第二,这道题有三个问题,第一个问题,最长,第二个问题,保留最长链,第三个问题,完成字典序。我们依次解决。
最长,可以使用树状DP,状态方程如下图所示,当前节点的最长链等于1+最长子链,靠dfs完成操作。
保存最长链,有两种方法,一种是动态遍历,等结束后再重新根据条件便利一次,输出结果长度,第二种方法,创建数组保存最长链,因为代码实现简单,本题创建son数组存储最长链。
字典序,通过条件判断来限制字典序,具体在代码中有展示,相同长度取字典序小的记录son中,也就完成了字典序筛选。