第八届蓝桥杯决赛 发现环 【dfs+并查集】

小明的实验室有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

每读取一条路,在起点终点相连之前,先对他们使用并查集判断是否属于同一父亲,如果两个点在相连之前便已经相连,则可以说明这两个店处于环中,我们可以将他们视为环的起点和终点。
然后用dfs找到起点到终点的路径。

#include 
using namespace std;

const int maxn=100000+5;

int fa[maxn];
bool vis[maxn];
int ans[maxn];

vector <int> e[maxn];

int n;
int st,ed;

void init(){
    memset(vis,0,sizeof(vis));
    for (int i=1;i<=n;i++){
        fa[i]=i;
    }
}

int find(int x){
    if (fa[x]==x){
        return x;
    }
    return fa[x]=find(fa[x]);
}

void unit(int x,int y){

    if (find(x)==find(y)){
        st=x;
        ed=y;
    }
    else{
        fa[x]=y;
    }
}


void dfs(int u,int id){
    vis[u]=true;
    ans[id]=u;

    if (u==ed){
        sort(ans,ans+id+1);
        for (int i=0;i<=id;i++){
            printf("%d%c",ans[i],i==id?'\n':' ');
        }
        return;
    }

    for (int i=0;iint v=e[u][i];
        if (!vis[v]){
            dfs(v,id+1);
        }
    }
}


int main(){

    cin >> n;

    init();
    for (int i=0;iint u,v;
        cin >> u >> v;
        unit(u,v);
        e[u].push_back(v);
        e[v].push_back(u);
    }
    dfs(st,0);
}

你可能感兴趣的:(DFS,并查集)