病毒容易发生变异。某种病毒可以通过突变产生若干变异的毒株,而这些变异的病毒又可能被诱发突变产生第二代变异,如此继续不断变化。
现给定一些病毒之间的变异关系,要求你找出其中最长的一条变异链。
在此假设给出的变异都是由突变引起的,不考虑复杂的基因重组变异问题 —— 即每一种病毒都是由唯一的一种病毒突变而来,并且不存在循环变异的情况。
输入格式:
输入在第一行中给出一个正整数 N(≤104),即病毒种类的总数。于是我们将所有病毒从 0 到 N−1 进行编号。
随后 N 行,每行按以下格式描述一种病毒的变异情况:
k 变异株1 …… 变异株k
其中 k 是该病毒产生的变异毒株的种类数,后面跟着每种变异株的编号。第 i 行对应编号为 i 的病毒(0≤i<N)。题目保证病毒源头有且仅有一个。
输出格式:
首先输出从源头开始最长变异链的长度。
在第二行中输出从源头开始最长的一条变异链,编号间以 1 个空格分隔,行首尾不得有多余空格。如果最长链不唯一,则输出最小序列。
注:我们称序列 { a1,⋯,an } 比序列 { b1,⋯,bn } “小”,如果存在 1≤k≤n 满足 ai=bi 对所有 i<k 成立,且 ak<bk。
输入样例:
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结点,需要去判断,没作为变异毒株出现过的才是树根结点
AC代码:
#include<bits/stdc++.h>
using namespace std;
vector<vector<int>> v(100010); //存储每个病毒可产生的变异毒株
vector<int> ans;//最终结果序列
void dfs(int root,vector<int> &road){
if(v[root].size() == 0){//到达叶子结点
if(road.size() > ans.size()){ //要更长的序列
ans = road;
}else if(road.size() == ans.size() && road < ans){ //序列长相同,选择更小的序列
ans = road;
}
return;
}
for(int i = 0;i < v[root].size();i++){
road.push_back(v[root][i]);
dfs(v[root][i],road);
road.pop_back();
}
}
int main(){
int N;
cin >> N;
bool book[N+1] = {false};
for(int i = 0;i < N;i++){
int K,x;
cin >> K;
for(int j = 0;j < K;j++){
cin >> x;
v[i].push_back(x);
book[x] = true;
}
}
int start;
for(int i = 0;i < N;i++){
if(!book[i]){
start = i;
break;
}
}
vector<int> cur;
cur.push_back(start);
dfs(start,cur);
int flag = 0;
cout << ans.size() << endl;
for(int i = 0;i < ans.size();i++){
if(flag++)
cout << " ";
cout << ans[i];
}
}
BFS做法:
#include<bits/stdc++.h>
using namespace std;
const int N = 1e4+10;
#define x first
#define y second
bool st[N];
int pa[N];
int n, start, l, tp;
vector<int> res;
vector<vector<int>> v(N);
typedef pair<int, int> PII;
queue<PII> q;
int main(){
cin >> n;
for(int i = 0;i < n;i++){
int m, x;
cin >> m;
while(m--) cin >> x, st[x] = true, v[i].push_back(x), pa[x] = i;
sort(v[i].begin(), v[i].end());
}
for(int i = 0;i < n;i++) if(st[i] == false) start = i;
q.push({start, 1});
while(!q.empty()){
int u = q.front().x, rd = q.front().y;
q.pop();
if(rd > l) l = rd, tp = u;
for(auto t : v[u]) q.push({t, rd + 1});
}
cout << l << endl;
while(tp != start) res.push_back(tp), tp = pa[tp];
res.push_back(start);
for(int i = res.size() - 1;i >= 0;i--){
cout << res[i];
if(i != 0) cout << " ";
}
}