DFS 树上搜索 T2 todo 病毒容易发生变异。某种病毒可以通过突变产生若干变异的毒株,而这些变异的病毒又可能被诱发突变产生第二代变异,如此继续不断变化。
现给定一些病毒之间的变异关系,要求你找出其中最长的一条变异链。
在此假设给出的变异都是由突变引起的,不考虑复杂的基因重组变异问题 —— 即每一种病毒都是由唯一的一种病毒突变而来,并且不存在循环变异的情况。
输入格式:
输入在第一行中给出一个正整数 ,即病毒种类的总数。于是我们将所有病毒从 到 进行编号。
随后 N$ 行,每行按以下格式描述一种病毒的变异情况:
k 变异株1 …… 变异株k
其中 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按序号从小到大搜索子树, 第一个等于最大深度的点就是答案。
不过可能是因为没看到根节点不一定为0导致WA, 后来改了一下思路, 用逐步更新的方式直接进行DFS, 若当前的长度大于之前的答案长度, 就更新长度并覆盖原来的答案序列。
依然是用 array 来代替指针数组, 直接一个=号处理掉更新序列。存图的话直接用邻接表vector, 只需要遍历的话用不着idx指针邻接表存图。
代码
#include <iostream>
#include <cstring>
#include <algorithm>
#include <string>
#include <cmath>
#include <vector>
#include <array>
using namespace std;
const int N = 1e4 + 10;
int n;
int d[N], root;
vector<int> g[N];
array<int, N> path, ans;
int max_cnt;
void dfs(int u, int cnt)
{
if(cnt > max_cnt)
{
path[cnt] = u;
ans = path;
max_cnt = cnt;
}
for(auto i : g[u])
{
path[cnt] = u;
dfs(i, cnt + 1);
}
}
int main()
{
cin >>n;
for(int i = 0; i < n; i++)
{
int m;
cin >> m;
for(int j = 0; j < m; j++)
{
int t;
cin >>t;
d[t]++;
g[i].push_back(t);
}
sort(g[i].begin(), g[i].end());
}
for(int i = 0; i < n;i ++) if(!d[i]) root = i;
dfs(root,0);
cout << max_cnt+1 << endl;
cout << ans[0];
for(int i = 1; i <= max_cnt; i++)
cout << " " << ans[i];
return 0;
}