0
点赞
收藏
分享

微信扫一扫

天梯赛:L2-038 病毒溯源 (25 分)

zhongjh 2022-03-13 阅读 100

题目详情 - 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);
}
举报

相关推荐

0 条评论