pat-病毒溯源

Posted 2021-05-18 Mr.wu123

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

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

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

输入格式：
输入在第一行中给出一个正整数 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
原题链接
查询从根节点到叶子节点的最长路径，如果存在多条输出字典数最小的一条。

#include<bits/stdc++.h>
using namespace std;//树形dp问题

const int N=10010,M=10010;
int n;
int h[N],e[M],ne[M],idx;
int son[N];
bool st[N];

void add(int a,int b){   //添加一条边a->b
   e[idx]=b,ne[idx]=h[a],h[a]=idx++;
}
int dfs(int u){
   int res=0;
   son[u]=-1;
   for(int i=h[u];~i;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]=min(son[u],j);
   }
   return res+1;
}

int main()
{
    memset(h,-1,sizeof h);
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        int cnt;
        scanf("%d",&cnt);
        while(cnt--){
            int x;
            scanf("%d",&x);
            add(i,x);
            st[x]=true;//i到x的边
        }
    }

    int root=0;
    while(st[root]) root++;//当前节点有父节点，往后走
    printf("%d\\n",dfs(root));//从根节点往下搜，找到最长路径
    printf("%d",root);
    while(son[root]!=-1){
        root=son[root];
        printf(" %d",root);
    }

    printf("\\n");
    cout<<h[0]<<endl;
    return 0;
}