联通森林2

文章描述了一种算法,用于解决给定森林结构中判断两个节点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。

样例

输入

6 3 2
1 2
1 3
2 1
1 2
1 4

输出

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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值