找朋友并查集裸题
找朋友
并查集用來(lái)合并兩個(gè)不相交的集合
主要找根,需要路徑優(yōu)化
合并的時(shí)候按秩合并
ac代碼
#include<bits/stdc++.h> using namespace std; const int maxn=5e4; int parent[maxn],rank1[maxn]; int m,n,p; struct frd{int x,y; }a[maxn],b[maxn];//找根 int find_root(int l,int parent[]){int l_root=l;while(parent[l_root]!=-1){l_root=parent[l_root];}return l_root; }//合并 int Union(int l ,int r,int parent[],int rank1[]){int x_root=find_root(l,parent);int y_root=find_root(r,parent);if(x_root==y_root) return 0;else{if(rank1[x_root]>rank1[y_root]){parent[y_root]=x_root;}else if(rank1[x_root]<rank1[y_root]){parent[x_root]=y_root;}else {parent[x_root]=y_root;rank1[y_root]++;}}return 1; } //判斷新來(lái)的是否是有一個(gè)祖先 bool judge(int l ,int r){int x_root=find_root(l,parent);int y_root=find_root(r,parent);if(x_root==y_root) return true;return false; }int main(){memset(parent,-1,sizeof(parent));//表示獨(dú)立的節(jié)點(diǎn)memset(rank1,0,sizeof(rank1));//秩,用來(lái)優(yōu)化樹(shù)的高度cin>>n>>m>>p;for(int i=0;i<m;i++){cin>>a[i].x>>a[i].y;int result=Union(a[i].x,a[i].y,parent,rank1); }for(int i=0;i<p;i++){cin>>b[i].x>>b[i].y;if(judge(b[i].x,b[i].y)) cout<<"Yes"<<endl;else cout<<"No"<<endl;}} 《新程序員》:云原生和全面數(shù)字化實(shí)踐50位技術(shù)專家共同創(chuàng)作,文字、視頻、音頻交互閱讀總結(jié)
- 上一篇: matlab2ask和2psk仿真实验代
- 下一篇: 广州的士官工资多少