并查集——Call to your teacher

并查集——Call to your teacher

题目描述

从实验室出来后，你忽然发现你居然把自己的电脑落在了实验室里，但是实验室的老师已经把大门锁上了。更糟的是，你没有那个老师的电话号码。你开始给你知道的所有人打电话，询问他们有没有老师的电话，如果没有，他们也会问自己的同学来询问电话号码。那么，你能联系到老师并且拿到电脑吗。

输入描述：

存在多组测试样例
每组样例的第一行分别是两个整数n(1<n<=50)，m(1<m<=2000)，n是在题目当中出现的人数，其中你的序号是1号，实验室老师的序号是n。
接下来的m行，每行有两个整数x(1<=x<=n)，y(1<=y<=n)，代表x有y的电话号码。

输出描述：

对于每组测试样例，如果你最终能联系到老师，输出“Yes”，否则输出“No”。

示例1

输入

5 5
1 3
2 3
3 4
2 4
4 5

输出

Yes

示例2

输入

4 3
1 2
2 3
4 1

输出

No

方法一：搜索+vector数组

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int maxn=5000;
vector<int> f[maxn];
int vis[maxn];
void bfs(int s)
{
	vis[s]=1;
	queue<int> q;
	q.push(s);
	while(!q.empty()){
		int p=q.front();
		q.pop();
		for(int i=0;i<f[p].size();i++){
			if(!vis[f[p][i]]){
				vis[f[p][i]]=1;
				q.push(f[p][i]);
			}
		}
	}
}
int main()
{
	int n,m;
	cin>>n>>m;
	memset(vis,0,sizeof(vis));
	for(int i=0;i<m;i++){
		int x,y;
		cin>>x>>y;
		f[x].push_back(y);
	}
	bfs(1);
	if(vis[n]==1){
		cout<<"Yes"<<endl;
	}
	else{
		cout<<"No"<<endl;
	}
	return 0;
}

方法二：并查集

这种方法要注意当输入的x等于n时，不能用并查集相连。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1e5+50;
int vis[maxn];
int find(int x)
{
	if(vis[x]==x){
		return x;
	}
	else{
		vis[x]=find(vis[x]);
		return vis[x];
	}
}
void merge(int a,int b)
{
	int p=find(a);
	int q=find(b);
	if(p==q){
		return ;
	}
	else{
		vis[q]=p;
		return ;
	}
}
int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=100000;i++){
		vis[i]=i;
	}
	for(int i=0;i<m;i++){
		int x,y;
		cin>>x>>y;
		if(x!=n)
			merge(x,y);
	}
	if(find(1)==find(n)){
		cout<<"Yes"<<endl;
	}
	else{
		cout<<"No"<<endl;
	}
	return 0;
}