在社交的过程中,通过朋友,也能认识新的朋友。在某个朋友关系图中,假定 A 和 B 是朋友,B 和 C 是朋友,那么A 和 C 也会成为朋友。即,我们规定朋友的朋友也是朋友。
现在,已知若干对朋友关系,询问你某两个人是不是朋友。
请编写一个程序来解决这个问题吧。
输入格式
第一行:三个整数 n,m,p (n\leq 5000,m\leq 5000,p\leq 5000)n,m,p(n≤5000,m≤5000,p≤5000),分别表示有 nn 个人,mm 个朋友关系,询问 pp 对朋友关系。
接下来 mm 行:每行两个数 A_i,B_iAi,Bi,1\leq A_i,B_i\leq N1≤Ai,Bi≤N,表示 A_iAi 和 B_iBi 具有朋友关系。
接下来 pp 行:每行两个数,询问两人是否为朋友。
输出格式
输出共 pp 行,每行一个Yes
或No
。表示第 ii 个询问的答案为是否朋友。
样例输入
6 5 3 1 2 1 5 3 4 5 2 1 3 1 4 2 3 5 6
样例输出
Yes Yes No
以上是题目
这道题是完完全全的不带权的并查集的题,做法就是基础的 查询合并,要注意的是 当两个高度不等的树合并时,小的往大的上面并!
最后附上代码和注释
#include <iostream> using namespace std; int n,m,p,a[10001],b[10001],par[10001],gao[10001]; //查找函数 int find(int x){ if(par[x] == x){ return x; }else { return par[x] = find (par[x]); } } //合并函数 void unite(int x,int y){ x = find(x); y = find(y); if(x==y) return; if(gao [x] < gao[y]){ par[x] = y; }else{ par[y] = x; if(gao[x] == gao[y]) gao[x]++; } } //判断函数 bool same(int x,int y){ return find(x) == find (y); } int main() { cin>>n>>m>>p; for(int i=0;i<n;i++) { par[i]=i; gao[i]=0; } for(int i=0;i<m;i++) { cin>>a[i]>>b[i]; unite(a[i],b[i]); } for(int i=0;i<p;i++) { int sa,sb; cin>>sa>>sb; if ( same(sa,sb) ) cout<<"Yes"<<endl; else cout<<"No"<<endl; } return 0; }