7-8 冗余连接 (25 分)
树可以看成是一个连通且 无环 的 无向 图。
给定往一棵 n 个节点 (节点值 1~n) 的树中添加一条边后的图。添加的边的两个顶点包含在 1 到 n 中间,且这条附加的边不属于树中已存在的边。
请找出一条可以删去的边,删除后可使得剩余部分是一个有着 n 个节点的树。如果有多个答案,则找到最后出现的边编号。
输入格式
第一行输入正整数n(n<=100)表示树节点的数量 随后n行,每行给出两个节点编号,表示该两点相连。
输出格式
输出可以删除的边的编号(按输入顺序,第i行给定的边编号为i),该边删除后可使得剩余部分是一个有着 n 个节点的树。如果有多个答案,则输出最后的边编号
输入样例
3
1 2
1 3
2 3
输出样例
3
基本思路
简单的并查集问题。
枚举i从1到n:读入两条边ans1、ans2,如果这两条边的根结点不一样就合并这两条边;如果碰到第一组根结点相同的情况,输出当前是第几组边i,返回。
代码
#include <bits/stdc++.h>
using namespace std;
int n;//边数
vector<int> father;//下标为顶点编号,值为该顶点的父顶点的编号
//查找
int findFather(int v){
if(father[v]==v){
return father[v];
}else{
//路径压缩
int F=findFather(father[v]);
father[v]=F;
return F;
}
}
//合并
void Union(int a,int b){
int faA=findFather(a);
int faB=findFather(b);
if(faA!=faB){
father[faA]=faB;
}
}
int main(){
cin>>n;
//初始化
father.resize(10005);
for(int i=1;i<=10005;i++) father[i]=i;
//读入所有边
for(int i=1;i<=n;i++){
int ans1,ans2;
cin>>ans1>>ans2;
if(findFather(ans1)==findFather(ans2)){
cout<<i<<endl;
return 0;
}else{
Union(findFather(ans1),findFather(ans2));
}
}
return 0;
}