联通森林2

coduck_lys

已于 2024-02-01 09:42:34 修改

阅读量297

点赞数 2

CC 4.0 BY-SA版权

文章标签：算法图论深度优先

于 2024-01-31 12:00:59 首次发布

本文链接：https://blog.youkuaiyun.com/coduck_lys/article/details/135950443

文章描述了一种算法，用于解决给定森林结构中判断两个节点a和b是否在同棵树上的问题，通过深度优先搜索（DFS）来确定节点的连通性。输入是森林的节点数量、边的数量和查询次数，输出为Yes或No。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

时间限制：1秒内存限制：128M

题目描述

输入若干森林，保证数据合法，请你判断某两个点是否在同一棵树上。

输入描述

第一行两个数n,m和q (0<=m<n<=1000000)，(1<=q<=10000)表示当前森林的节点个数为n(节点编号从1开始，n结束)，以及m条边，接下来m行，每行x和y,表示x和y节点有连边，注：数据中可能存在重边.。接下来q行，每行两个数a和b，表示询问a和b是否位于同一棵树上。

输出描述

对于q中的每个询问，如果a和b属于同一棵树，则输出Yes，否则输出No。

样例

输入

输出

Yes
No

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
using namespace std;
int n,m,q,a,b;
int x,y;
const int N=1e6+10;
int head[N*2],vis[N*2],ver[N*2],Next[N*2],tot=-1,cnt;
void Add(int x,int y) {
	ver[++tot]=y;
	Next[tot]=head[x];
	head[x]=tot;
}
void dfs(int pos) {
	vis[pos]=cnt;
	for(int i=head[pos]; i!=-1; i=Next[i]) {
		int ljd=ver[i];
		if(vis[ljd]==0) {
			dfs(ljd);
		}
	}
}

int main() {
	memset(head,-1,sizeof(head));
	cin>>n>>m>>q;
	for(int i=1; i<=m; i++) {
		cin>>x>>y;
		Add(x,y);
		Add(y,x);
	}
	for(int i=1; i<=n; i++) {
		if(vis[i]==0) {
			cnt++;
			dfs(i);
		}
	}
	while(q--){
		cin>>a>>b;
		if(vis[a]==vis[b]){
			cout<<"Yes"<<endl;
		}
		else{
			cout<<"No"<<endl;
		}
	}
	return 0;
}