P1551 亲戚 深度优先搜索题解

P1551 亲戚
这里给出一个深度优先遍历的算法。O(n)的预处理，每个回答O(1)。

#include<iostream>
#include<vector>
using namespace std;
vector <int> g[5001];
int d[5001];
int cnt=1;
void dfs(int x)//预处理，求每个结点所在的集合编号
{
	d[x]=cnt;//x所所在集合的编号，通过这个遍历我们可以求出与x关系的结点在同一个编号中
	for (int i=0;i<g[x].size();i++)
	{
		if (d[g[x][i]]==0)
		   dfs(g[x][i]);
	}
}
int main()
{
	int n,m,k;
	cin>>n>>m>>k;
	for (int i=0;i<m;i++)
	{
		int x,y;
		cin>>x>>y;
		g[x].push_back(y);
		g[y].push_back(x);
	}
	for (int i=1;i<=n;i++)
	{
		if (d[i]==0)//对每个结点求集合编号
		{
			dfs(i);
			cnt++;
		}
	}
	for (int i=1;i<=k;i++)
	{
		int x,y;
		cin>>x>>y;
		if (d[x]==d[y])//编号相同为一个集合
		   cout<<"Yes"<<endl;
		else
		   cout<<"No"<<endl;
	}
}
/*
样例输入 
6 5 3
1 2
1 5
3 4
5 2
1 3
1 4
2 3
5 6
样例输出
Yes
Yes
No 
*/