发现环 拓扑排序 DFS

问题 1841: [蓝桥杯][2017年第八届真题]发现环

时间限制: 1Sec 内存限制: 128MB 提交: 610 解决: 216

题目描述

小明的实验室有N台电脑，编号1~N。原本这N台电脑之间有N-1条数据链接相连，恰好构成一个树形网络。在树形网络上，任意两台电脑之间有唯一的路径相连。
不过在最近一次维护网络时，管理员误操作使得某两台电脑之间增加了一条数据链接，于是网络中出现了环路。环路上的电脑由于两两之间不再是只有一条路径，使得这些电脑上的数据传输出现了BUG。
为了恢复正常传输。小明需要找到所有在环路上的电脑，你能帮助他吗？

输入

第一行包含一个整数N。
以下N行每行两个整数a和b，表示a和b之间有一条数据链接相连。
对于30%的数据，1 <= N <= 1000
对于100%的数据, 1 <= N <= 100000， 1 <= a, b <= N
输入保证合法。

输出

按从小到大的顺序输出在环路上的电脑的编号，中间由一个空格分隔。

样例输入

5
1 2
3 1
2 4
2 5
5 3

样例输出

1 2 3 5

思路1：

因为这个图是一棵树加了一条边，形成了一个环，那么我们利用拓扑排序，不断取出度数为1或可以变为1的节点，然后留下的就是环里面的点。

代码1：

#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=100000+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
int n;
int du[maxn];
vector<int>edge[maxn];
bool vis[maxn];
int main()
{
    cin>>n;
    rep(i,1,n)
    {
        int u,v;
        cin>>u>>v;
        edge[u].push_back(v);
        edge[v].push_back(u);
        du[u]++;
        du[v]++;
    }
    queue<int>q;
    rep(i,1,n)
    {
        if(du[i]==1)
        {
           q.push(i);
           vis[i]=1;
        }
    }
    while(q.size())
    {
        int x=q.front();
        q.pop();
        int s=edge[x].size();
        rep(i,0,s-1)
        {
            int v=edge[x][i];
            du[v]--;
            if(du[v]==1)
            {
                vis[v]=1;
                q.push(v);
            }
        }
    }
    rep(i,1,n)
    {
        if(!vis[i])cout<<i<<" ";
    }
    cout<<endl;
    return 0;
}

思路2：

利用dfs遍历这个图，每次遍历时记录这个点的前驱，遇到环时输出。

代码2：

#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=100000+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
int n;
vector<int>edge[maxn];
int pre[maxn];
int ans[maxn];
bool vis[maxn];
int tot;
int flag;
void dfs(int x,int fa)
{
    int s=edge[x].size();
    for(int i=0; i<s; i++)
    {
        int v=edge[x][i];
        if(v==fa)
            continue;
        pre[v]=x;//记录v的前驱是x
        if(vis[v])
        {
            //出现环了
            int tmp=v;
            flag=1;
            while(true)
            {
                tmp=pre[tmp];
                ans[++tot]=tmp;
                if(tmp==v)
                    break;
                //cout<<tmp<<endl;
            }
            return ;
        }
        else
        {
            vis[v]=1;
            dfs(v,x);
        }
        if(flag)
            return ;
    }
}
int main()
{
    cin>>n;
    rep(i,1,n)
    {
        int u,v;
        cin>>u>>v;
        edge[u].push_back(v);
        edge[v].push_back(u);
    }
    vis[1]=1;
    dfs(1,0);//从1点开始搜索
    sort(ans+1,ans+tot+1);
    rep(i,1,tot)
    {
        cout<<ans[i]<<" ";
    }
    cout<<endl;
    return 0;
}