学习资料：代码随想录
108. 冗余连接
卡码网题目链接（ACM模式）
判断是否有环的依据为，利用并查集，isSame函数，判断当下这条边的两个节点入集前是否为同根，如果是的话，该边就是会构成环的那条边
#include <iostream>
#include <vector>
using namespace std;
int n;
vector<int> father=vector<int>(1001,0);
void init(){
for(int i=0;i<n;i++){
father[i]=i;
}
}
int find(int u){
return u==father[u]?u:father[u]=find(father[u]);
}
bool isSame(int u,int v){
u=find(u);
v=find(v);
return u==v;
}
void join(int u,int v){
u=find(u);
v=find(v);
if(u==v) return;
father[v]=u;
}
int main(){
int s,t;
cin>>n;
init();
for(int i=0;i<n;i++){
cin>>s>>t;
if(isSame(s,t)) {
cout<<s<<' '<<t<<endl;
return 0;
}
else{
join(s,t);
}
}
}109. 冗余连接II
卡码网题目链接（ACM模式）
分三种情况：代码随想录
对于有两个入度的节点的情况，需要先倒序记录节点入度情况，找到入度为2的那个节点对应的两条边，判断应该删除哪条边。判断方法为，先看第一条（之前是倒序遍历，即看最后这条），用isSame函数判段是否为构成环的这条边，如果是，cout这条，如果不是cout另一条;
如果没有入度为2的节点，即情况3，直接用isSame函数判断就可以了
#include <iostream>
#include <vector>
using namespace std;
int n;
vector<int> father=vector<int>(1001,0);
void init(){
for(int i=1;i<=n;i++){
father[i]=i;
}
}
int find(int u){
return u==father[u]?u:father[u]=find(father[u]);
}
bool isSame(int u,int v){
u=find(u);
v=find(v);
return u==v;
}
void join(int u,int v){
u=find(u);
v=find(v);
if(u==v) return;
father[v]=u;
}
bool isTreeAfterDeleteEdge(const vector<vector<int>>& edges,int todelete){
init();
for(int i=0;i<n;i++){
if(i==todelete){
continue;
}
if(isSame(edges[i][0],edges[i][1])) return false;
join(edges[i][0],edges[i][1]);
}
return true;
}
void deletedEdge(const vector<vector<int>>& edges){
init();
for(int i=0;i<n;i++){
if(isSame(edges[i][0],edges[i][1])){
cout<<edges[i][0]<<' '<<edges[i][1];
return;
}
join(edges[i][0],edges[i][1]);
}
}
int main(){
cin>>n;
vector<vector<int>> edges;
vector<int> inDegree(n+1,0);
int s,t;
for(int i=0;i<n;i++){
cin>>s>>t;
edges.push_back({s,t});
inDegree[t]++;
}
vector<int> doubleDegree;
for(int i=n-1;i>=0;i--){
if(inDegree[edges[i][1]]==2){
doubleDegree.push_back(i);
}
}
if(doubleDegree.size()>0){
if(isTreeAfterDeleteEdge(edges,doubleDegree[0])){
cout<<edges[doubleDegree[0]][0]<<' '<<edges[doubleDegree[0]][1];
}
else{
cout<<edges[doubleDegree[1]][0]<<' '<<edges[doubleDegree[1]][1];
}
return 0;
}
deletedEdge(edges);
}有几个小点：
1、既然 isTreeAfterRemoveEdge() 和 getRemoveEdge() 都是靠并查集判断是否成环，能不能只用 getRemoveEdge() 一个函数来处理所有情况？
不能！
getRemoveEdge() 只能处理「无入度为 2」的情况（即情况三），
如果你碰到的是「有一个点入度为 2」（即情况一、情况二），就必须要先判断删哪一条边，这一步 getRemoveEdge() 做不到。
2、对于find函数会改变并查集的结构，这并不改变每一个节点的根节点，所以虽然在join函数和isSame函数的使用中，find会使并查集结构改变，对判断没有影响
3、对于vector，赋值方法v[i] = x需要以前设置vector大小，赋值方法v.push_back(x)不需要提前设置vector大小
