题目详情 - L2-038 病毒溯源 (25 分) (pintia.cn)
就是一个简单的图的遍历加dfs,我的方法就是利用并查集找到根节点,在从根节点开始进行dfs。
可怜的蒟蒻不知道怎么存储路径,最后参照这位大佬的做法才完成了这道题天梯赛L2-38 病毒溯源 (25 分)_m0_51385035的博客-CSDN博客
就是搞不懂为啥根节点可以这么算。。。
#include<cstdio>
#include<set>
#include<vector>
#include<algorithm>
using namespace std;
const int maxn=1e4+1;
vector<set<int>> s(maxn);
vector<int> v,c;
void dfs(int pos)
{
for(auto i:s[pos])
{
v.emplace_back(i);
dfs(i);
if(s[i].empty()&&c.size()<v.size())
c=v;
v.pop_back();
}
}
int main()
{
int n,i,k,j,a,b[maxn],t,d;
scanf("%d",&n);
for(i=0;i<n;i++)
b[i]=i;
for(i=0;i<n;i++)
{
scanf("%d",&k);
for(j=0;j<k;j++)
{
scanf("%d",&a);
s[i].insert(a);
t=i;
b[a]=i;
}
}
while(b[t]!=t)
t=b[t];
c.emplace_back(t);
v=c;
dfs(t);
printf("%d\n",c.size());
for(i=0;i<c.size()-1;i++)
printf("%d ",c[i]);
auto it=s[c[i-1]].begin();
printf("%d",*it);
}
自己搞得测试数据,天知道我在这里卡了多久......
10
2 6 4
0
1 8
0
2 5 9
0
1 7
1 2
0
2 3 1