时间限制: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;
}